diff --git a/norch/__init__.py b/norch/__init__.py
index c2a3122..80da58b 100644
--- a/norch/__init__.py
+++ b/norch/__init__.py
@@ -3,6 +3,7 @@ from .nn import *
 from .optim import *
 from .utils import *
 from .norchvision import *
+from . import data
 
 __version__ = "0.0.4"
 __author__ = 'Lucas de Lima Nogueira'
diff --git a/norch/utils/data/__init__.py b/norch/utils/data/__init__.py
index 2397554..e5ea955 100644
--- a/norch/utils/data/__init__.py
+++ b/norch/utils/data/__init__.py
@@ -1,4 +1,5 @@
 from .dataset import *
 from .example import *
 from .dataloader import *
-from .batch import *
\ No newline at end of file
+from .batch import *
+from .distributed import *
\ No newline at end of file