Editing shortcode output