Create problem_5.py