Update hw6-1.py