tweaks to the demo