Re-jig styles and add registering shortcode