Add undo states visualization