Update hw1p3.py