updated the heatmap to use the heatmap data type object