added if statement on the shortcode in the constructclasses