diff --git a/Project_Model/Libs/Transformer/Classes/__init__.py b/Project_Model/Libs/Transformer/Classes/__init__.py
index e6c69ae..837ea82 100644
--- a/Project_Model/Libs/Transformer/Classes/__init__.py
+++ b/Project_Model/Libs/Transformer/Classes/__init__.py
@@ -3,11 +3,13 @@ from .Encoder import Encoder
 from .FeedForwardNetwork import FeedForwardNetwork
 from .MultiHeadAttention import MultiheadAttention
 from .TorchMultiHeadAttention import TorchMultiHeadAttention
+from .SpannedMasker import SpannedMasker
 
 __all__ = [
     "Decoder",
     "Encoder",
     "FeedForwardNetwork",
     "MultiheadAttention",
-    "TorchMultiHeadAttention"
+    "TorchMultiHeadAttention",
+    "SpannedMasker"
 ]
\ No newline at end of file
diff --git a/Project_Model/Libs/Transformer/Enums/__init__.py b/Project_Model/Libs/Transformer/Enums/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/Project_Model/Libs/Transformer/Utils/__init__.py b/Project_Model/Libs/Transformer/Utils/__init__.py
new file mode 100644
index 0000000..e69de29