Add solution problem 2