Updated the demo notebook