Fix some small issues on the demo page