Re-add demo notebook to the repo