My problem 4 solution