heatmap now computes probability