fixed problem 3