Added prob1.py