hw2 part 2 fix