Add shortcode handling (callback functions)