Create problem_2.py