Update Wk4_Ex3.py