Update a1.py