Add update undo/redo button!