diff --git a/Local-Without-Prediction/docker-compose.yml b/Local-Without-Prediction/docker-compose.yml
index e3eff020e2a77ae2ddfe2a05f2174a7b2f93a26d..95565c9796d953ec024ad17de4eaea9eb17608bf 100644
--- a/Local-Without-Prediction/docker-compose.yml
+++ b/Local-Without-Prediction/docker-compose.yml
@@ -2,7 +2,7 @@ version: '3.8'
 services:
 
   node0:
-    image: local-comm
+    image: image-dfl-bad
     container_name: node0
     environment:
       - RANK=0
@@ -15,7 +15,7 @@ services:
 
 
   node1:
-    image: local-comm
+    image: image-dfl-bad
     container_name: node1
     environment:
       - RANK=1
@@ -28,7 +28,7 @@ services:
 
 
   node2:
-    image: local-comm
+    image: image-dfl-bad
     container_name: node2
     environment:
       - RANK=2
@@ -41,7 +41,7 @@ services:
 
 
   node3:
-    image: local-comm
+    image: image-dfl-bad
     container_name: node3
     environment:
       - RANK=3
diff --git a/Local-Without-Prediction/generate-docker-compose.py b/Local-Without-Prediction/generate-docker-compose.py
index 817bd1626ed31655f8bb90e5711390e578f18212..a54ec4c5fdf40b50d253d16fba02a138c0259688 100644
--- a/Local-Without-Prediction/generate-docker-compose.py
+++ b/Local-Without-Prediction/generate-docker-compose.py
@@ -1,8 +1,9 @@
 import sys
+import argparse
 
-image = 'local-comm'
+#image = 'image-dfl-bad'
 
-def generate_docker_compose(num_nodes):
+def generate_docker_compose(image_name, num_nodes):
     docker_compose_template = """\
 version: '3.8'
 services:
@@ -17,7 +18,7 @@ services:
       - BASE_PORT=8000
       - IP_ADDRESS=node{i}
     volumes:
-      - ./neighbors/neighbors{file_suffix}.txt:/app/neighbors.txt
+      - ./neighbors/neighbors{i}.txt:/app/neighbors.txt
     networks:
       - my_network
 
@@ -31,9 +32,8 @@ networks:
 
 
     docker_compose_content = docker_compose_template
-    for i in range(0, num_nodes):
-        file_suffix = i
-        docker_compose_content += service_template.format(i=i, file_suffix=file_suffix,img=image)
+    for i in range(0, int(num_nodes)):
+        docker_compose_content += service_template.format(i=i,img=image_name)
 
     docker_compose_content += network_section
 
@@ -41,9 +41,17 @@ networks:
         file.write(docker_compose_content)
 
 if __name__ == "__main__":
-    if len(sys.argv) != 2:
-        print("Usage: python generate_docker_compose.py <num_nodes>")
-        sys.exit(1)
 
-    num_nodes = int(sys.argv[1])
-    generate_docker_compose(num_nodes)
+    parser = argparse.ArgumentParser(description="Upload PDF files to an S3 bucket")
+    parser.add_argument("--image_name", help="The name of the image")
+    parser.add_argument("--nb_nodes", help="The number of nodes")
+    args = parser.parse_args()
+    # image_name= args.image_name
+    # if len(sys.argv) != 2:
+    #     print("Usage: python generate_docker_compose.py <num_nodes>")
+    #     sys.exit(1)
+
+    # num_nodes = int(sys.argv[1])
+    generate_docker_compose(args.image_name, args.nb_nodes)
+
+