Add Undo/Redo controller back