Update and rename prob2 to prob2.py