hw problem 2 fixed