diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..b59f7e3
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+test/
\ No newline at end of file
diff --git a/CountFolderFiles.py b/CountFolderFiles.py
index ff7c9bd..9ffdb24 100644
--- a/CountFolderFiles.py
+++ b/CountFolderFiles.py
@@ -27,16 +27,7 @@ def count_folder_files(folder_path):
 def check_directory_exists(directory_name):
     return os.path.isdir(directory_name)
 
-def main():
-    if len(sys.argv) != 2:
-        print("Usage: python script.py <folder_path>")
-        sys.exit(1)
-
-    directory_name = sys.argv[1]
-    suffix = count_folder_files(directory_name)
-
-    # If [num] already exists at the end of the string, remove it so we can update it
-
+def clean_directory_name(directory_name):
     # Find the last occurrence of backslash
     last_backslash_index = directory_name.rfind('\\')
     
@@ -45,11 +36,20 @@ def main():
     
     # If '[' exists after the last backslash, remove everything after it
     if last_open_bracket_index != -1:
-        directory_name_cleaned = directory_name[:last_open_bracket_index]
+        return directory_name[:last_open_bracket_index]
     else:
         # If '[' doesn't exist, keep the original string
-        directory_name_cleaned = directory_name
+        return directory_name
 
+def main():
+    if len(sys.argv) != 2:
+        print("Usage: python script.py <folder_path>")
+        sys.exit(1)
+
+    directory_name = sys.argv[1]
+    suffix = count_folder_files(directory_name)
+    # If [num] already exists at the end of the string, remove it so we can update it
+    directory_name_cleaned = clean_directory_name(directory_name)
     
     if check_directory_exists(directory_name):
         print(f"Directory '{directory_name}' exists.")