In ANH, one of the ways we are shown Luke’s kindness is by having him treat droid’s like living beings.
In ROTJ, one gets scared that Luke only sees them as disposable tools when he gives them to Jabba, only for him to make sure to retrieve them later, showing he was never gonna leave them behind.

I think it would cool if in the first film, you can establish Anakin as having a similar kindness towards droids, but by the third film, make it clear that he views them as tools. It’s a good way to show a gradual psychological turn towards Vader, as one of Vader’s defining traits is his dehumanization and objectification of everyone around him.
Extend this to his troops. Anakin wanting to go back for the clones should be in episode 2. By episode 3, he’s so desensitized he cares only for the mission.