more work on the visualizations and stuff