Update HW5.py