diff --git a/scripts/ml/ml_iterator.py b/scripts/ml/ml_iterator.py
index 7231db4..0832708 100755
--- a/scripts/ml/ml_iterator.py
+++ b/scripts/ml/ml_iterator.py
@@ -41,21 +41,26 @@ gene_model_paramD = {'data_combined_model'       : False
 ###############################################################################
 #ml_genes = ["pncA", "embB", "katG", "rpoB", "gid"]
 
-ml_gene_drugD = {'pncA'   : 'pyrazinamide'
-                 # , 'embB' : 'ethambutol'
-                 # , 'katG' : 'isoniazid'
-                 # , 'rpoB' : 'rifampicin'
-                 # , 'gid'  : 'streptomycin'
+ml_gene_drugD = {
+        #'pncA'   : 'pyrazinamide'
+        'embB'   : 'ethambutol'
+        , 'katG' : 'isoniazid'
+        , 'rpoB' : 'rifampicin'
+        , 'gid'  : 'streptomycin'
                  }
 gene_dataD={}
-split_types = ['70_30',
-               '80_20',
-               'sl',
-               'rt',
-               'none_bts']
+split_types = [
+        '70_30',
+        '80_20',
+        'sl',
+        #'rt',
+        #'none_bts'
+               ]
 
-split_data_types = ['actual',
-                    'complete']
+split_data_types = [
+        'actual',
+        #'complete'
+        ]
 
 for gene, drug in ml_gene_drugD.items():
     print ('\nGene:', gene