Added data transformation for heatmap