added heatmap visualization