Create 001_problem.py