Python solution to problem 14