Updated Step4.ipynb