updating hw1.py