Updated test notebook.