Made the solution to problem 20 more compact.