Create prob3.py