Update notebook.
Fix plotting code.
Update notebook