Fixed visualization so that proper colors display for edges and vertices.