got colors working for the 3D visualization