Update hw2p3.py