added additional visualizations and graph