small update to demo notebook