diff --git a/scripts/ml/ml_functions/TEST_GetMLData.py b/scripts/ml/ml_functions/TEST_GetMLData.py
index 040af65..be1c14b 100644
--- a/scripts/ml/ml_functions/TEST_GetMLData.py
+++ b/scripts/ml/ml_functions/TEST_GetMLData.py
@@ -5,6 +5,7 @@ Created on Thu Sep  1 12:22:27 2022
 
 @author: tanu
 """
+
 getmldata(gene = "alr"
           , drug = "cycloserine"
           , data_combined_model = False
diff --git a/scripts/ml/ml_iterator.py b/scripts/ml/ml_iterator.py
index 0832708..b8a0a42 100755
--- a/scripts/ml/ml_iterator.py
+++ b/scripts/ml/ml_iterator.py
@@ -40,10 +40,9 @@ gene_model_paramD = {'data_combined_model'       : False
                     , 'write_outfile'            : False }
 ###############################################################################
 #ml_genes = ["pncA", "embB", "katG", "rpoB", "gid"]
-
 ml_gene_drugD = {
-        #'pncA'   : 'pyrazinamide'
-        'embB'   : 'ethambutol'
+        'pncA'   : 'pyrazinamide'
+        , 'embB' : 'ethambutol'
         , 'katG' : 'isoniazid'
         , 'rpoB' : 'rifampicin'
         , 'gid'  : 'streptomycin'
@@ -53,13 +52,13 @@ split_types = [
         '70_30',
         '80_20',
         'sl',
-        #'rt',
-        #'none_bts'
+        'rt',
+        'none_bts'
                ]
 
 split_data_types = [
-        'actual',
-        #'complete'
+        #'actual',
+        'complete'
         ]
 
 for gene, drug in ml_gene_drugD.items():