add clustering functionality to heatmap plot