Add solution for problem 8