Update HW9.py