Update hw2.py

all finished up
last commit before due