Create problem 9.py