Update Wk4_Ex2.py