draw_heat function

Draws the heatmap.