Update notebook.py