Add solution for problem 5