Researchers call for better diagnosis and treatment of mental illness for offenders in prison and after release.