Update p1.06.py