minor changes to the visualization