Update hw2.py

Updated! sorry that was kinda a silly mistake!