diff --git a/scripts/functions/tests/test_dm_om_data.R b/scripts/functions/tests/test_dm_om_data.R
index d1391cf..06f848b 100644
--- a/scripts/functions/tests/test_dm_om_data.R
+++ b/scripts/functions/tests/test_dm_om_data.R
@@ -7,7 +7,11 @@ source("~/git/LSHTM_analysis/config/gid.R")
 #source("~/git/LSHTM_analysis/config/rpob.R")
 
 source("~/git/LSHTM_analysis/scripts/plotting/get_plotting_dfs.R")
-
+########################################################################
+  geneL_na      = c("gid", "rpob")
+  geneL_dy      = c("gid")
+  geneL_ppi2    = c("alr", "embb", "katg", "rpob")
+########################################################################
 all_dm_om_df = dm_om_wf_lf_data(df = merged_df3, gene_name = gene)
 
 wf_duet      = all_dm_om_df[['wf_duet']]
@@ -31,12 +35,17 @@ lf_consurf   = all_dm_om_df[['lf_consurf']]
 wf_snap2     = all_dm_om_df[['wf_snap2']]
 lf_snap2     = all_dm_om_df[['lf_snap2']]
 
+if (tolower(gene)%in%geneL_na){
 wf_mcsm_na   = all_dm_om_df[['wf_mcsm_na']]
 lf_mcsm_na   = all_dm_om_df[['lf_mcsm_na']]
+}
 
+if (tolower(gene)%in%geneL_ppi2){
 wf_mcsm_ppi2 = all_dm_om_df[['wf_mcsm_ppi2']]
 lf_mcsm_ppi2 = all_dm_om_df[['lf_mcsm_ppi2']]
+}
 
+if (tolower(gene)%in%geneL_dy){
 wf_dynamut   = all_dm_om_df[['wf_dynamut']]
 lf_dynamut   = all_dm_om_df[['lf_dynamut']]
 
@@ -50,4 +59,5 @@ wf_sdm       = all_dm_om_df[['wf_sdm']]
 lf_sdm       = all_dm_om_df[['lf_sdm']]
 
 wf_mcsm      = all_dm_om_df[['wf_mcsm']]
-lf_mcsm      = all_dm_om_df[['lf_mcsm']]
\ No newline at end of file
+lf_mcsm      = all_dm_om_df[['lf_mcsm']]
+}
diff --git a/scripts/plotting/get_plotting_dfs.R b/scripts/plotting/get_plotting_dfs.R
index 01c21be..5864ebe 100644
--- a/scripts/plotting/get_plotting_dfs.R
+++ b/scripts/plotting/get_plotting_dfs.R
@@ -127,7 +127,7 @@ cat(s1)
 ####################################################################
 #source("other_plots_data.R")
 # converted to a function
-# moved old one to redundant. Added suffix to filename i.e. _nf(non-function)
+# moved old one to redundant.
 source(paste0(plot_script_path, "dm_om_data.R"))
  
 s2 = c("\nSuccessfully sourced other_plots_data.R")