added some visualizations