adding heatmap :D