Assignment1Soln_v2.py