a couple changes in the demo notebook