Updated Step2.ipynb