Create Problem 2.py