adding the missing heatmap code file