Policy to improve mental health reform in the United States criminal justice system

If you could suggest 1 policy to improve mental health reform in the criminal justice system, what would it be? Why?