80% done a3.py