Add shortcode class to functions.php