added comments in the demo notebook.