grading hw3.py