Create Problem 3.py