Add option nb colors to heatmap