Update hw1p1.py