Officials at the Supermax federal prison in Florence are taking steps to improve treatment of inmates with mental illness.