Update hw3.py