Added a heatmap!