fixed problem 3 code