diff --git a/scripts/functions/corr_plot_df.R b/scripts/functions/corr_plot_df.R
index 37b6b49..4cb878b 100644
--- a/scripts/functions/corr_plot_df.R
+++ b/scripts/functions/corr_plot_df.R
@@ -1,10 +1,27 @@
+#!/usr/bin/env Rscript  
+#########################################################
+# TASK: Script to format data for Correlation plots: 
+# corr_data_extract()
+# Input:
+    # corr_plot_df: data with all parameters (my_use case)
+        # merged_df3 or merged_df2!?
+    # gene: [sanity check]
+    # drug: relates to a column name that will need to extracted
+   
 #colnames_to_extract = c("mutationinformation"
  #                         , "duet_affinity_change")
 #display_colnames_key = c(mutationinformation = "MUT"
 #                           , duet_affinity_change = "DUET")
+# extract_scaled_cols = T or F, so that parameters with the _scaled suffix can be extracted.
+# No formatting applied to these cols i.e display name
+
+# TO DO: SHINY
+    #1) Corr type?
+    #2)
+
 ##################################################################
 corr_data_extract <- function(corr_plot_df
-                              , gene_name = gene
+                              #, gene_name = gene
                               , drug_name = drug
                               , colnames_to_extract
                               , colnames_display_key
@@ -18,7 +35,7 @@ corr_data_extract <- function(corr_plot_df
         , "\n=========================================")
     
     cat("\nExtracting default columns for"
-        , "\nGene name:", gene
+        #, "\nGene name:", gene
         , "\nDrug name:", drug)
     
     colnames_to_extract =  c(drug
@@ -26,7 +43,8 @@ corr_data_extract <- function(corr_plot_df
                              , "mutation_info_labels"  
                              , "duet_stability_change" 
                              , "ligand_affinity_change"
-                             , "ligand_distance"
+                             #, "ligand_distance"
+                             , LigDist_colname
                              , "ddg_foldx"
                              , "deepddg"
                              , "asa"
@@ -59,7 +77,8 @@ corr_data_extract <- function(corr_plot_df
     # arg: colnames_display_key
     colnames_display_key =  c(duet_stability_change  = "DUET"
                             , ligand_affinity_change = "mCSM-lig"
-                            , ligand_distance        = "ligand_distance"
+                            #, ligand_distance        = "ligand_distance"
+                            #, LigDist_colname        = "ligand_distance"
                             , ddg_foldx              = "FoldX"
                             , deepddg                = "DeepDDG"
                             , asa                    = "ASA"