diff --git a/bin/make_heatmap.R b/bin/make_heatmap.R
index f76d419f171a6f07667a229ea449c5e441d9f689..4f4eb60a265bb463e57b80f25a4dc83daa0748b5 100644
--- a/bin/make_heatmap.R
+++ b/bin/make_heatmap.R
@@ -23,6 +23,12 @@ setkey(trait_name, "ID")
 Rho = fread(correlation_file)
 setkey(Rho, "V1")
 
+D = as.matrix(1-Rho[,-1])
+D[is.na(D)] = 0
+h = hclust(as.dist(D))
+order_trait <- trait_name$ID[h$order]
+
+
 Pval = fread(p_file)
 setkey(Pval, "V1")