I also have a theory that part of the need for this belief in punishment is "there have to be bad people who deserve it" --> "because it means they are worse than me" --> "which makes me a good person who can go to heaven (while other people go to hell)"