Updated the example notebook.