diff --git a/scripts/ml/ml_iterator.py b/scripts/ml/ml_iterator.py
index 6daf527..ea9002c 100755
--- a/scripts/ml/ml_iterator.py
+++ b/scripts/ml/ml_iterator.py
@@ -50,11 +50,12 @@ ml_gene_drugD = {'pncA'   : 'pyrazinamide'
 gene_dataD={}
 split_types = ['70_30',
                '80_20',
-               'sl'
-               ]
+               'sl',
+               'rt',
+               'none_bts']
+
 split_data_types = ['actual',
-                    'complete'
-                    ]
+                    'complete']
 
 for gene, drug in ml_gene_drugD.items():
     print ('\nGene:', gene