Add solution to problem 5