fixed problem 2