Add color map to surface plot