plot_heatmap function

Create a heatmap