solution to problem 16 in python