Update problem_5.py