Create problem 2.py