Adds new style to css and some minor changes into demo module