Update prog_v1.py