Add TODO in the undo function