Added problem 17 in python