Heatmap now working

Heatmap working with weighting.