diff --git a/scripts/ml/ml_functions/GetMLData.py b/scripts/ml/ml_functions/GetMLData.py
index a61b0da..bd2d703 100755
--- a/scripts/ml/ml_functions/GetMLData.py
+++ b/scripts/ml/ml_functions/GetMLData.py
@@ -57,7 +57,10 @@ def getmldata(gene, drug
                           , 'dist_lineage_proportion'
                           #, 'lineage' # could be included as a category but it has L2;L4  formatting
                           , 'lineage_count_all'
-                          , 'lineage_count_unique']
+                          , 'lineage_count_unique'
+                          , 'snp_frequency' # NEW
+                          , 'df2_pos_count_all'# NEW
+                          , 'pos_count'] # NEW
     
     # X_gn_Fcat = ['drtype_mode_labels'  # beware then you can't use it to predict [USED it for uq_v1, not v2]
     #                #, 'gene_name'] # will be required for the combined stuff
diff --git a/scripts/ml/ml_iterator_CVs.py b/scripts/ml/ml_iterator_CVs.py
old mode 100644
new mode 100755
index 233f6fb..19db5f2
--- a/scripts/ml/ml_iterator_CVs.py
+++ b/scripts/ml/ml_iterator_CVs.py
@@ -15,7 +15,8 @@ homedir = os.path.expanduser("~")
 sys.path.append(homedir + '/git/LSHTM_analysis/scripts/ml/ml_functions')
 sys.path
 ###############################################################################
-outdir = homedir + '/git/LSHTM_ML/output/genes/'
+#outdir = homedir + '/git/LSHTM_ML/output/genes/'
+outdir = homedir + '/git/Misc/'
 
 #====================
 # Import ML functions