Add solution for problem 7