Update notebook.ipynb