diff --git a/Part 2/main.py b/Part 2/main.py
index cf8836e0efd52dddb89f97e98dde59689c831070..3448e4a4d5be5019f80a7a82f3dbd306d99d5450 100644
--- a/Part 2/main.py	
+++ b/Part 2/main.py	
@@ -127,8 +127,8 @@ def main(endpoint, index_name):
 if __name__== "__main__":
     # Argument parsing
     parser = argparse.ArgumentParser(description='Configure endpoint and index name for the lecture chat application.')
-    parser.add_argument('endpoint', type=str, help='The endpoint for the OpenSearch service.')
-    parser.add_argument('index_name', type=str, help='The index name for storing embeddings.')
+    parser.add_argument('--endpoint', type=str, help='The endpoint for the OpenSearch service.')
+    parser.add_argument('--index_name', type=str, help='The index name for storing embeddings.')
     args = parser.parse_args()
     main(args.endpoint, args.index_name)