Split off the visualization into it's own branch