--- a/create_heatmap.R	Wed Nov 15 08:58:11 2017 -0500
+++ b/create_heatmap.R	Fri Nov 17 09:41:07 2017 -0500
@@ -16,6 +16,7 @@
     l = dim(data_matrix)[1];
     p = (sqrt(9 + 8 * (k - 1)) - 3) / 2;
     m = as.matrix(data_matrix[,1+1:p]/data_matrix[,1]);
+    colnames(m) = colnames(data_matrix)[1+1:p];
     marks = colnames(m);
     rownames(m) = paste(1:l-1," (", round(data_matrix[,1] / sum(data_matrix[,1]) * 10000) / 100, "%)", sep="");
     pdf(file=opt$output);