Update problem_008.py