Add heatmap experiment