Update prog2.py