A few minor changes to demo page