Fixed colorizing in mayavi