Hi Jamie - I'm not really sure where I would move the "Remove" button without it feeling really weird, it is kind of naturally paired with "Add".

But I think that a special undo that worked while you were in the Image command is feasible and should solve that problem.

- Michael