trying again to fix heatmap