Create Problem 5.py