diff --git a/src/dirmanager.py b/src/dirmanager.py
index e5a395e..4fa4844 100644
--- a/src/dirmanager.py
+++ b/src/dirmanager.py
@@ -2,19 +2,33 @@ from pathlib import Path
 
 
 class DirManager:
-    def __init__(self, root_tests_dir: Path, session_id: str):
+    """Manages directories for the tests and should be used to create and find
+    and use the correct directories.
+
+    The structures is as follows:
+    tests dir/
+        session_dir-1/
+            records
+            states
+            results
+        session_dir-2/
+            records
+            ...
+    """
+
+    def __init__(self, tests_dir: Path, session_id: str):
         # root test dir
-        self.root_tests_dir = root_tests_dir
+        self.tests_dir = tests_dir
         self.session_id = session_id
 
         self.get_all_dirs()
 
     def get_all_dirs(self):
-        self.session_dir = self.root_tests_dir / f"test-{self.session_id}"
+        self.session_dir = self.tests_dir / f"test-{self.session_id}"
         self.session_sub_dirs = self.get_subdirs(self.session_dir)
 
     def create_all_dirs(self):
-        self.create_dirs(self.root_tests_dir, exist_ok=True)
+        self.create_dirs(self.tests_dir, exist_ok=True)
         self.create_dirs(self.session_dir)
         self.create_dirs(self.session_sub_dirs)
 
diff --git a/src/main.py b/src/main.py
index 322814f..cfda325 100644
--- a/src/main.py
+++ b/src/main.py
@@ -42,7 +42,7 @@ class Wrapper:
         self.run_test()
 
     def setup_test(self, session_id: str):
-        self.dir_manager = DirManager(root_tests_dir=TESTS_DIR, session_id=session_id)
+        self.dir_manager = DirManager(tests_dir=TESTS_DIR, session_id=session_id)
         self.dir_manager.create_all_dirs()
 
     def run_test(self):