hide the z-sigma for 2d data
correct transfer labels for 2D data