Minor improvements to demo page