improvements for the visualization