diff --git a/code/infrastructure/streaming/clients/sub/memgraph/Dockerfile b/code/infrastructure/streaming/clients/sub/memgraph/Dockerfile
index a1b4a9428b679eb831adb72b05591bd794e23f6c..377cd39c12545784b226dcfe68691af690a57ed9 100644
--- a/code/infrastructure/streaming/clients/sub/memgraph/Dockerfile
+++ b/code/infrastructure/streaming/clients/sub/memgraph/Dockerfile
@@ -16,7 +16,8 @@ RUN pip install neo4j
 
 # Copy the Python script into the container
 #COPY sub_mem.py /app/
-COPY sub_mem_whole_batch_subgraph_oriented.py /app/
+# COPY sub_mem_whole_batch_subgraph_oriented.py /app/
+COPY ./kind_oriented/sub_mem_whole_batch_kind_oriented_two_label_internal_collision_handle.py /app/
 
 # Set environment variable 'time_limit'
 ENV abort_time_limit=999999
@@ -26,4 +27,4 @@ ENV mos_port=1883
 ENV create_indices=False
 
 # Run the Python script
-CMD ["python", "sub_mem_whole_batch_subgraph_oriented.py"]
+CMD ["python", "sub_mem_whole_batch_kind_oriented_two_label_internal_collision_handle.py"]
diff --git a/code/infrastructure/streaming/clients/sub/memgraph/Notes_and_legend.txt b/code/infrastructure/streaming/clients/sub/memgraph/Notes_and_legend.md
similarity index 100%
rename from code/infrastructure/streaming/clients/sub/memgraph/Notes_and_legend.txt
rename to code/infrastructure/streaming/clients/sub/memgraph/Notes_and_legend.md
index 2c4318a0f8eac50fd9fdd74d08dc5484c5899d5f..5f87b6a8bac2ce1813233f6bb11a8a4eef56e49e 100644
--- a/code/infrastructure/streaming/clients/sub/memgraph/Notes_and_legend.txt
+++ b/code/infrastructure/streaming/clients/sub/memgraph/Notes_and_legend.md
@@ -6,6 +6,7 @@ Unwind = UNWIND
 no label = one label for all nodes and the original label as a property of the node.
 
 sub_mem_original = subscriber from the original experiment, added comment at line 15 and 32
+
 These comments describes what was changed for all new subscribers
 Note: in general the changes are not explicitly marked, but it should be possible to identify all 
       changed locations by a diff program or ide function. 
@@ -15,11 +16,10 @@ I encountered Pylint towards the end of this period.
 The most severe messages generated by Pylint were addressed.
 Others like the layout recommendations have not been implemented due to lack of time.
 
-
 Procedure_for_module_in_memgraph = is the procedure 
 
-
 The newest and best version at the moment is 'sub_mem_whole_batch_kind_oriented_two_label_internal_collision_handle'.
+
 Using 'OPTIONAL MATCH' throws an error at the edge, which tries to create an edge on a null node. 
 This null node wasn't found by 'OPTIONAL MATCH' a normal 'MATCH' would stop the query, but it woudln't throw an error.
 So, this Version is good for identifying cases that need to be handled.