Added styles and demo functionality