Add heatmap lib