hw1.py final