Create problem-1.py