Update hw2.py
Merge pull request #1 from cascam07/master

Update hw2.py