Some improvements to the visualization.