Update hw6.py