Added heatmap

With colors n shit