Solution to Problem 20
Title added
Cosmetic changes