Minor tweak to Problem 4 Ruby solution.