Minor changes to notebook.