Treatment, not punishment