Undo/Redo buttons work

Also enabled, disabled correctly