updated hw2.py to do this hw