docs update for heatmap wrapper function