fixed problem 4