; and (B) for the remaining 36-month period of such project, provide to each State selected under paragraph (4) payments in accordance with paragraph (5). (2) Purpose The purpose described in this paragraph is for each State selected under paragraph (4) to increase the treatment capacity of providers participating under the State plan (or a waiver of such plan) to provide substance use disorder treatment or recovery services under such plan (or waiver) through the following activities: (A) For the purpose described in paragraph (3)(C)(i), activities that support an ongoing assessment of the behavioral health treatment needs of the State, taking into account the matters described in subclauses (I) through (IV) of such paragraph. (B) Activities that, taking into account the results of the assessment described in subparagraph (A), support the recruitment, training, and provision of technical assistance for providers participating under the State plan (or a waiver of such plan) that offer substance use disorder treatment or recovery services. (C) Improved reimbursement for and expansion of, through the provision of education, training, and technical assistance, the number or treatment capacity of providers participating under the State plan (or waiver) that— (i) are authorized to dispense drugs approved by the Food and Drug Administration for individuals with a substance use disorder who need withdrawal management or maintenance treatment for such disorder; (ii) have in effect a registration or waiver under section 303(g) of the Controlled Substances Act for purposes of dispensing narcotic drugs to individuals for maintenance treatment or detoxification treatment and are in compliance with any regulation promulgated by the Assistant Secretary for Mental Health and Substance Use for purposes of carrying out the requirements of such section 303(g)