Add solution to problem 20