Add a toggle for moving an entity on undo