[Feature] Add undo and redo button