Update hw2p5.py