Adding some minor fixes

Adding some minor fixes to the demo page