Add JS logic and shortcode