It'd be awesome if `Esc` closed the visualization, it took me a moment to figure out I had to click the icon again to close out of it.