Update a2q2.py