Updated notebook.ipynb