made some more changes to heatmap