diff --git a/openai_server/agent_tools/driverless_ai_data_science.py b/openai_server/agent_tools/driverless_ai_data_science.py
index 8e6904b5b..7fc083003 100644
--- a/openai_server/agent_tools/driverless_ai_data_science.py
+++ b/openai_server/agent_tools/driverless_ai_data_science.py
@@ -418,7 +418,7 @@ def main():
     parser = argparse.ArgumentParser(description="Run Driverless AI experiments from command line.")
 
     # instance
-    parser.add_argument("--engine", "--dai_engine", default="daidemo",
+    parser.add_argument("--engine", "--dai_engine", default=os.getenv('DAI_ENGINE', "daidemo"),
                         help="Name of the DAI engine")
     parser.add_argument("--client_id", "--dai_client_id", default=os.getenv('DAI_CLIENT_ID', "hac-platform-public"),
                         help="Name of client_id")
@@ -448,7 +448,7 @@ def main():
                         help="Source type of the dataset (default: s3)")
 
     # Creating new experiment
-    parser.add_argument("--target-column",
+    parser.add_argument("--target-column", "--target",
                         default="Churn?",
                         required=False,
                         help="Name of the target column for prediction")
diff --git a/src/version.py b/src/version.py
index 0a86202e4..9efbe6fc2 100644
--- a/src/version.py
+++ b/src/version.py
@@ -1 +1 @@
-__version__ = "b437c5a5e6089ff42c2fb556b20df0b0a32dcece"
+__version__ = "711ddb361d5ce3b694628bab066a2537e38f1143"