Add solution problem 5