Update assgn1_3.py