hw2 part3 fix
hw2 part3 fix