Create Problem 6.py