hw2 part2 fix