Create prob7.py