made the visualization better hopefully