diff --git a/procedural_city_generation/roadmap/config_functions/input_image_setup.py b/procedural_city_generation/roadmap/config_functions/input_image_setup.py
index 2c60353..94a30c8 100644
--- a/procedural_city_generation/roadmap/config_functions/input_image_setup.py
+++ b/procedural_city_generation/roadmap/config_functions/input_image_setup.py
@@ -17,22 +17,10 @@ def input_image_setup(rule_image_name, density_image_name):
     density_img: np.ndarray
         Density-image as numpy array
     '''
-
-    #TODO: Document
-    import matplotlib.image as mpimg
-    import procedural_city_generation
-    import os
-    path=os.path.dirname(procedural_city_generation.__file__)
-
-    rule_img = mpimg.imread(path+"/inputs/rule_pictures/"+rule_image_name)
-    density_img = mpimg.imread(path+"/inputs/density_pictures/"+density_image_name)
-
-    import matplotlib.pyplot as plt
-    plt.imsave(path+"/temp/"+density_image_name.split(".")[0]+"diffused.png", density_img, cmap='gray')
-    with open(path+"/temp/"+density_image_name.split(".")[0]+"isdiffused.txt", 'w') as f:
-        f.write("False")
-
-
-    rule_img*=255
-    density_img*=255
-    return rule_img, density_img
+    # Load the rule image
+    rule_img = cv2.imread(rule_image_name, cv2.IMREAD_GRAYSCALE)
+    # Load the density image
+    density_img = cv2.imread(density_image_name, cv2.IMREAD_GRAYSCALE)
+    # Save the density image in temp folder
+    cv2.imwrite('temp/density_image.png', density_img)
+    return rule_img, density_img
\ No newline at end of file