Expanding on your question... there are four purposes of punishment in penology: deterrence, incapacitation, retribution and rehabilitation.
I added a little video here that I made a while ago for further explanation.