Minor changes on the demo page