Made hw2 changes