diff --git a/llvm/projects/soc_simulator/src/driver.py b/llvm/projects/soc_simulator/src/driver.py
index 23b7fe26b91c68f2cbaba9b0bb853295f2cb21df..b685170da40c17dc45b7258cbbe2166ac52736fc 100644
--- a/llvm/projects/soc_simulator/src/driver.py
+++ b/llvm/projects/soc_simulator/src/driver.py
@@ -276,13 +276,18 @@ def display_results(results_filename):
         results_file.write("\nc%d,%f\n\n" % (best_config, aggregate_results[attribute][best_config]))
     results_file.close()
 
+def driver(tensor_layers_file, tensor_table_file, conf_file, output_file):
+    print(tensor_layers_file, tensor_table_file, conf_file, output_file)
+    parse_tensor_layer_file(tensor_layers_file)
+    parse_tensor_table(tensor_table_file)
+    run_simulations(conf_file)
+    display_results(output_file)
+
 if __name__ == "__main__":
-    '''
-    if len(sys.argv) != 4):
+    if len(sys.argv) != 5:
         print("Usage: python driver.py <layer info> <tensor info> <configurations> <results file>")
         exit(1)
-    '''
-    parse_tensor_layer_file("/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_mobilenet/mobilenet_layers.txt")
-    parse_tensor_table("/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_pldi/mobilenet_results/mobilenet_tensors.txt")
-    run_simulations("/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_mobilenet/mobilenet_conf2.txt")
-    display_results("blah.txt")
+    test_layers_file = "/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_mobilenet/mobilenet_layers.txt"
+    test_table_file = "/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_pldi/mobilenet_results/mobilenet_tensors.txt"
+    test_conf_file = "/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_mobilenet/mobilenet_conf2.txt"
+    driver(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4])