added visualizations and made slight modifications