; or (iii) are qualified under applicable State law to provide substance use disorder treatment or recovery services. (D) Improved reimbursement for and expansion of, through the provision of education, training, and technical assistance, the number or treatment capacity of providers of substance use disorder treatment or recovery services participating under the State plan (or waiver), including providers that have the qualifications to address the treatment or recovery needs of— (i) individuals enrolled under the State plan (or a waiver of such plan) who have neonatal abstinence syndrome, in accordance with guidelines issued by the American Academy of Pediatrics and American College of Obstetricians and Gynecologists relating to maternal care and infant care with respect to neonatal abstinence syndrome; (ii) pregnant women, postpartum women, and infants, particularly the concurrent treatment, as appropriate, and comprehensive case management of pregnant women, post­partum women and infants, enrolled under the State plan (or a waiver of such plan); (iii) adolescents and young adults between the ages of 12 and 21 enrolled under the State plan (or a waiver of such plan); or (iv) American Indian and Alaska Native individuals enrolled under the State plan (or a waiver of such plan). (3) Planning grants (A) In general The Secretary shall, with respect to the first 18-month period of the demonstration project conducted under paragraph (1), award planning grants to at least 10 States selected in accordance with subparagraph (B) for purposes of preparing an application described in paragraph (4)(C) and carrying out the activities described in subparagraph (C). (B) Selection In selecting States for purposes of this paragraph, the Secretary shall— (i) select States that have a State plan (or waiver of the State plan) approved under this title; (ii) select States in a manner that ensures geographic diversity