Weighted heatmaps working