Update assignment0/assignment0.py