assignment-2c.py