; (ii) by redesignating subsections (e) through (i) as subsections (f) through (j), respectively; and (iii) by inserting after subsection (d) the following: (e) Additional use of funds A State may use an additional 2 percent of the funds annually allotted to the State under this section for qualified nonprofit small treatment works technical assistance providers (as the term is defined in section 222) and qualified nonprofit medium treatment works technical assistance providers (as the term in defined in section 223) to provide technical assistance to small treatment works (as the term is defined in section 222) and medium treatment works (as the term is defined in section 223) in the State. . (2) Conforming amendment Section 221(d) of the Federal Water Pollution Control Act ( 33 U.S.C. 1301(d) ) is amended by striking section 603(h) and inserting section 603(i) . 5005. Clean, safe, reliable water infrastructure (a) Drinking water infrastructure (1) Other authorized activities Section 1452(k) of the Safe Drinking Water Act ( 42 U.S.C. 300j–12(k) ) is amended— (A) in paragraph (1)(D), by inserting and the implementation of plans to protect source water identified in a source water assessment under section 1453 before the period at the end