Create problem 7.py