heatmap is now a bit faster