Update prog.py