Improved the look of the demo