Update prog002.py