updated a4.py