Add solution for problem 10