Problem 26
Minor tweak