Create problem1.py