try to make this heatmap code work