Why do you think punishment is only about rehabilitation?

Punishment has three purposes, namely, retribution, rehabilitation, and deterrence.

I also don't know how much rehabilitation actually occurs in our prisons. From what I understand, not much.