fixed problem 1