> prison for punishment, not rehab.

Prisons are already an idealistic institution, in the interest of being more humane. The alternative before was capital punishment, work camps, or slavery (including in war). That's not to say prisons are ok in their current state, or can't be improvement. I just don't think rehab was historically the primary intent.