Create 006_problem.py