diff --git a/scripts/ml/functions/GetMLData.py b/scripts/ml/functions/GetMLData.py
index cfd762a..f8155d9 100755
--- a/scripts/ml/functions/GetMLData.py
+++ b/scripts/ml/functions/GetMLData.py
@@ -609,8 +609,11 @@ def getmldata(gene, drug
 
     print('\n#################################################################'
           , '\nSUCCESS: Extacted training data for gene:', gene.lower()
-          , '\nDim of training_df:', all_training_df.shape
-          , '\nThis includes OR'
+          , '\nDim of training_df:', all_training_df.shape)
+    if use_or:
+        print('\nThis includes Odds Ratio')
+    else:
+        print('\nThis EXCLUDES Odds Ratio'
           , '\n###############################################################')
 
     #==========================================================================