Add solution to problem 40