works with heatmap 2.0!