I woulda done away with the whole “fear of loss leads to evil” or whatever. Instead, I’d rather that Anakin have a god complex and think that he needs to inherit as much power as possible to maintain order across the galaxy. I think that aligns more with the Empire’s values anyways. I would also probably have Obi-Wan be the central character of the PT, rather than Anakin, and have Anakin’s downfall happen at the end of the second movie, and then the third movie is all about Obi-Wan coming to terms with things.
A god complex would be intriguing, there is a foundation for that in AOTC with Anakin conversation with Padme:
Padme: You are not all powerful Ani.
Anakin: Well I should be. Someday I will be. I will become the most powerful Jedi ever! I promise you. I will even learn how to stop people from dying.
In those lines Anakin says he wants omnipotence like a god, and to control death is god like too.