Update problem_016.py