Remove debugging cruft
Fix undo and add quick visualization fn