First version of heatmap