Why must it be complicated? Punishment should equal the crime, yes? Good behavior and rehabilitation, and such should be weighed against the judgement, the scales of justice must still be balanced, yes? It goes with the natural law of action to reaction.
Who is the judge, where is the courtroom, and where is the prison is a matter we can only speculate. Personally, I believe God is judge, the courtroom is in heaven, and the prison is here on earth. But it's only my deduction as of today.
Yes, agree that punishment should equal the crime
Adam's deliberate crime was breaking the Law that carried with it the death penalty - Genesis 2:17
No post-mortem penalty for Adam, No double jeopardy to be paid by Adam, none for anyone else
There is only one (1) punishment for sin and that (1) punishment is: death
Not death plus anything else - Romans 6:23,7 - Nothing past death - Ecclesiastes 9:5; Psalm 115:17; Isaiah 38:18
The reason for the first prophecy (Gen. 3:15 about the promised 'seed' aka Messiah) is to balance the Scales of Justice
Both Adam and Jesus started out sinless. Unlike us, they could only sin on purpose intentionally
Adam chose to break the law, Jesus chose to keep the law
Humanly perfect Adam sinned under favorable conditions
Humanly perfect Jesus did Not sin under adverse conditions
God is the Judge but God gave all judging to be done by faithful Jesus - John 5:22
So, at the soon coming 'time of separation' on Earth then Jesus as King is the Judge between the 'sheep&goats'
- Matthew 25:31-34,37
In this coming judgement case: the courtroom is Earth, those who are judged as wicked their 'prison' so to speak, is everlasting death aka being 'destroyed forever' - Psalms 92:7; 104:35; 145:20 Proverbs 2:21-22
The humble meek aka figurative sheep can inherit the Earth as Jesus promised at Matt. 5:5 from Psalms 37:9-11;22:26