Update hw1.py