Draft implementation of heatmap