sarting to write heatmap code