treatment vs prison