Update u2_2.py