Setting up a basic heatmap