hw2 part3 fix