Added the add_shortcode function