Cleaned up visualizations and output a bit.