problem 2 inital solution