problem_1.py