Their punishment is, as decided by law, being separated from society, family, friends, their hobbies and comfort while still getting to live d dignified human beings.
Not everyone in prison is a murderer or rapist, and even many of those get reformed. So please think of that before you go and approve of their exploitation.