fixed more inconsistencies in the visualization part