Update a2.py

Finished Coursera HW2