Update hw2.py

### added at bottom
Merge pull request #1 from lauton04/master

Update hw2.py