Added HeatMapData class