diff --git a/Makefile b/Makefile
index 7c1d8e0..7c9c61b 100644
--- a/Makefile
+++ b/Makefile
@@ -12,11 +12,11 @@ test-shapley:
 	$(TIME) python qii.py -m shapley final.csv
 
 test:
-	python qii.py -m average-unary-individual final.csv
-	python qii.py -m unary-individual final.csv
-	python qii.py -m discrim final.csv
-	python qii.py -m banzhaf final.csv
-	python qii.py -m shapley final.csv
+	$(TIME) python qii.py -m average-unary-individual final.csv
+	$(TIME) python qii.py -m unary-individual final.csv
+	$(TIME) python qii.py -m discrim final.csv
+	$(TIME) python qii.py -m banzhaf final.csv
+	$(TIME) python qii.py -m shapley final.csv
 
 clean:
 	rm -Rf *.pyc
diff --git a/ml_util.py b/ml_util.py
index faadf4e..758ab1e 100644
--- a/ml_util.py
+++ b/ml_util.py
@@ -277,9 +277,9 @@ def get_arguments():
     parser.add_argument('-c', '--classifier', default='logistic', help='Classifier to use',
             choices=['logistic', 'svm', 'decision-tree', 'decision-forest'])
 
-    parser.add_argument('--max_depth', default=2, help='Max depth for decision trees and forests')
-    parser.add_argument('--n_estimators', default=20, help='Number of trees for decision forests')
-    parser.add_argument('--seed', default=None, help='Random seed, auto seeded if not specified', type=int)
+    parser.add_argument('--max_depth', type=int, default=2, help='Max depth for decision trees and forests')
+    parser.add_argument('--n_estimators', type=int, default=20, help='Number of trees for decision forests')
+    parser.add_argument('--seed', type=int, default=None, help='Random seed, auto seeded if not specified', type=int)
     
     parser.add_argument('-i', '--individual', default=0, type=int, help='Index for Individualized Transparency Report')
     parser.add_argument('-r', '--record-counterfactuals', action='store_true', help='Store counterfactual pairs for causal analysis')