adding hw1.py