add problem 15 in python