Added age_group param to diagnoes functions