Create problem2.py