Updated A.py