small tweaks to demo