Add solution problem 3