Moving shortcode to a class