Minor modification to shortcode output