I would also love to see visitation alongside all of this (if that’s been brought up already I missed it, I just woke up). Lots of opportunity for crushingly sad interaction between jailed characters and their loved ones.
Expanding the prison system would, I think, be a net positive. More severe offenses (that don’t instantly warrant death like straight-up treason) could see characters jailed more often than killed, which would be an interesting new direction for a character’s story as opposed to an ending.