Updated shortcode code