updated the demo notebook