grading hw2.py