Create prob4.py