Yeah, that'd be great to add; I wish I had baked the undoing into the basic implementation to make things easier later.