Update demo.ipynb