3d plot working, with colormap
legend