There is something satisfying with ending with Vader’s helmet being sealed and the breathing, I will give you that. It is a good ending note.
I would highly suggest using the alternate scenes that depicts Padme being alive with Leia … it should be consistent with Leia’s statement in Return of the Jedi.
Granted, story wise the idea of Padme’s death would be better … her death, Vader’s birth … But the Prequels should follow along the original’s stories/canon, not the other way around.