make_heatmap function

Make main heatmap part of plot for gghm.