add_heatmap function

Add heatmap