Minor edits to notebook.