diff --git a/src/deep_image_matching/__main__.py b/src/deep_image_matching/__main__.py
deleted file mode 100644
index e69de29b..00000000
diff --git a/src/deep_image_matching/graph.py b/src/deep_image_matching/graph.py
index 32d7af40..9d4237bd 100644
--- a/src/deep_image_matching/graph.py
+++ b/src/deep_image_matching/graph.py
@@ -10,6 +10,8 @@
 
 logger = logging.getLogger("dim")
 
+TEMPLATE_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "utils/templates")
+
 
 def view_graph(db, output_dir, imgs_dir):
     logger.info("Creating view graph visualization...")
@@ -23,8 +25,7 @@ def view_graph(db, output_dir, imgs_dir):
     nt = Network()
 
     # HTML template for view graph details panel
-    current_directory = os.path.dirname(os.path.realpath(__file__))
-    nt.set_template(os.path.join(current_directory, "templates", "template.html"))
+    nt.set_template(os.path.join(TEMPLATE_DIR, "template.html"))
 
     # Add nodes
     G = nx.Graph()
diff --git a/src/deep_image_matching/templates/animation_template.html b/src/deep_image_matching/utils/templates/animation_template.html
similarity index 100%
rename from src/deep_image_matching/templates/animation_template.html
rename to src/deep_image_matching/utils/templates/animation_template.html
diff --git a/src/deep_image_matching/templates/lib/bindings/utils.js b/src/deep_image_matching/utils/templates/lib/bindings/utils.js
similarity index 100%
rename from src/deep_image_matching/templates/lib/bindings/utils.js
rename to src/deep_image_matching/utils/templates/lib/bindings/utils.js
diff --git a/src/deep_image_matching/templates/lib/tom-select/tom-select.complete.min.js b/src/deep_image_matching/utils/templates/lib/tom-select/tom-select.complete.min.js
similarity index 100%
rename from src/deep_image_matching/templates/lib/tom-select/tom-select.complete.min.js
rename to src/deep_image_matching/utils/templates/lib/tom-select/tom-select.complete.min.js
diff --git a/src/deep_image_matching/templates/lib/tom-select/tom-select.css b/src/deep_image_matching/utils/templates/lib/tom-select/tom-select.css
similarity index 100%
rename from src/deep_image_matching/templates/lib/tom-select/tom-select.css
rename to src/deep_image_matching/utils/templates/lib/tom-select/tom-select.css
diff --git a/src/deep_image_matching/templates/lib/vis-9.0.4/vis-network.css b/src/deep_image_matching/utils/templates/lib/vis-9.0.4/vis-network.css
similarity index 100%
rename from src/deep_image_matching/templates/lib/vis-9.0.4/vis-network.css
rename to src/deep_image_matching/utils/templates/lib/vis-9.0.4/vis-network.css
diff --git a/src/deep_image_matching/templates/lib/vis-9.0.4/vis-network.min.js b/src/deep_image_matching/utils/templates/lib/vis-9.0.4/vis-network.min.js
similarity index 100%
rename from src/deep_image_matching/templates/lib/vis-9.0.4/vis-network.min.js
rename to src/deep_image_matching/utils/templates/lib/vis-9.0.4/vis-network.min.js
diff --git a/src/deep_image_matching/templates/lib/vis-9.1.2/template_style.css b/src/deep_image_matching/utils/templates/lib/vis-9.1.2/template_style.css
similarity index 100%
rename from src/deep_image_matching/templates/lib/vis-9.1.2/template_style.css
rename to src/deep_image_matching/utils/templates/lib/vis-9.1.2/template_style.css
diff --git a/src/deep_image_matching/templates/lib/vis-9.1.2/vis-network.css b/src/deep_image_matching/utils/templates/lib/vis-9.1.2/vis-network.css
similarity index 100%
rename from src/deep_image_matching/templates/lib/vis-9.1.2/vis-network.css
rename to src/deep_image_matching/utils/templates/lib/vis-9.1.2/vis-network.css
diff --git a/src/deep_image_matching/templates/lib/vis-9.1.2/vis-network.min.js b/src/deep_image_matching/utils/templates/lib/vis-9.1.2/vis-network.min.js
similarity index 100%
rename from src/deep_image_matching/templates/lib/vis-9.1.2/vis-network.min.js
rename to src/deep_image_matching/utils/templates/lib/vis-9.1.2/vis-network.min.js
diff --git a/src/deep_image_matching/templates/template.html b/src/deep_image_matching/utils/templates/template.html
similarity index 100%
rename from src/deep_image_matching/templates/template.html
rename to src/deep_image_matching/utils/templates/template.html