Minor tweaks to demo menu