Add heatmap function to stats tools.