Updated the a4.py