Update problem_7.py