added visualizations and code