Punishment

Who invented the idea that, if you do something wrong, something bad should happen to you?  I mean, after you realize it's wrong and know why you did it, I thought the "punishment" was over.