Improvements to the demo page