Add solution problem 4