Undo and redo shortcuts working properly now