Add heatmap example