Update a2.py