; (ii) improving assessments used to determine the needs of the infant and family; (iii) improving ongoing case management services; and (iv) improving access to treatment services, which may be prior to the pregnant woman’s due date. (2) Developing policies, procedures, or protocols in consultation and coordination with health professionals, public and private health facilities, and substance use disorder treatment agencies to ensure that— (A) appropriate notification to child protective services is made in a timely manner; (B) a plan of safe care is in place, where needed, before the infant is discharged from the birth or health care facility; and (C) such health and related agency professionals are trained on how to follow such protocols and are aware of the supports that may be provided under a plan of safe care. (3) Training health professionals and health system leaders, child welfare workers, substance use disorder treatment agencies, and other related professionals such as home visiting agency staff and law enforcement in relevant topics including— (A) State mandatory reporting laws and the referral and notification process; (B) the co-occurrence of pregnancy and substance use disorder; (C) the clinical guidance about treating substance use disorder in pregnant and post­par­tum women