Update hw1p2.py