Hated the courtroom scene at the end. So, in real life...no one is flipping back on 7 years and blowing it up to get 87 in the courtroom. It took me out the scene.
Everything in the Breaking Bad universe wouldn't possibly happen in real life.
That said, you completely misunderstood the whole point in the character arcs of Jimmy to Saul back to Jimmy. There are two times where Jimmy was first disbarred and 2nd in hiding as Gene where he was miserable in not doing what he was. Him as being Saul gave him LIFE. However, being Saul got him to lose everything who he cared about, and Kim was the last straw. He had NOTHING out there. It's that realization that of either being in PRISON & happen to be finally FREE in being just Jimmy or being FREE but still feel like he's in prison like he was when he was Gene.