Create prob1.py