use new heatmap code