Add shortcode functionality