updated demo notebook