Create problem-001.py