Create problem_6.py