Update problem_010.py