Update hw1.py
Merge pull request #3 from linzac03/master

Update hw1.py