Improve styling of demo page