trying too solve heatmap again