Update q1.py