update problem_15.py