update hw1.py