Tweaks to demo page