Add initial shortcode functionality