Updated demo notebook.