add undo button, small fixes