fixed problem 4 part 1