Update hw2_2.py