more updates to demo notebook
demo notebook update