Update hw9.py