hw1.py