Add solution to problem 020