Create Problem 4.py