Minor update to the Python solution to problem 26