Add undo record on initial startup
Use another way to save tree state for undo