Add sample annotations to heatmap