Create prob2.py