diff --git a/pyproject.toml b/pyproject.toml
index 22a0e6d..9121c6c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,5 +1,5 @@
 [project]
-name = "speckle-automate-data-shield"
+name = "data-shield"
 version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
diff --git a/src/__init__.py b/src/data-shield/__init__.py
similarity index 100%
rename from src/__init__.py
rename to src/data-shield/__init__.py
diff --git a/src/actions.py b/src/data-shield/actions.py
similarity index 99%
rename from src/actions.py
rename to src/data-shield/actions.py
index 10056e9..f770611 100644
--- a/src/actions.py
+++ b/src/data-shield/actions.py
@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 from collections import defaultdict
-from typing import Dict, List, Optional, Union, Any
+from typing import Dict, List, Optional
 from speckle_automate import AutomationContext
 from specklepy.objects import Base
 
diff --git a/src/function.py b/src/data-shield/function.py
similarity index 98%
rename from src/function.py
rename to src/data-shield/function.py
index 0c40686..4a49161 100644
--- a/src/function.py
+++ b/src/data-shield/function.py
@@ -1,6 +1,6 @@
 from speckle_automate import AutomationContext
 from specklepy.objects import Base
-from actions import PrefixRemovalAction
+from src.actions import PrefixRemovalAction
 from src.inputs import FunctionInputs, SanitizationMode
 from traversal import get_data_traversal_rules
 
diff --git a/src/helpers.py b/src/data-shield/helpers.py
similarity index 100%
rename from src/helpers.py
rename to src/data-shield/helpers.py
diff --git a/src/inputs.py b/src/data-shield/inputs.py
similarity index 100%
rename from src/inputs.py
rename to src/data-shield/inputs.py
diff --git a/src/rule_processor.py b/src/data-shield/rule_processor.py
similarity index 100%
rename from src/rule_processor.py
rename to src/data-shield/rule_processor.py
diff --git a/src/rules.py b/src/data-shield/rules.py
similarity index 100%
rename from src/rules.py
rename to src/data-shield/rules.py
diff --git a/src/traversal.py b/src/data-shield/traversal.py
similarity index 100%
rename from src/traversal.py
rename to src/data-shield/traversal.py