I would just keep the Mustafar lines honestly. It’s not a huge deal if he doesn’t go straight there, and I even like how it’d be Anakin lying more explicitly to Padme.
How are you planning to handle the transitions? I might even want to try this in my own edit…