Update HW3.py