Minor updates to notebook.