Create problem_1.py