Visual tweaks to demo