Create 002_problem.py