Small update to the notebook.