improve the look of the demo