diagnosis and each major type of treatment service listed under subparagraph (B). (E) The number and percentage of individuals enrolled under the State plan under this title or the State child health plan under title XXI (or under a waiver of such plans) by major enrollment category, who have a diagnosis described in subparagraph (A) and received substance use disorder or mental health treatment through— (i) a Medicaid managed care entity (as defined in section 1932(a)(1)(B)), including the number of such individuals who received such assistance through a prepaid inpatient health plan (as defined by the Secretary) or a prepaid ambulatory health plan (as defined by the Secretary)