Add 'baby_stage' shortcode