diff --git a/lazuardy_anfis/__init__.py b/lazuardy_anfis/__init__.py
index 5820dff..7c3be6a 100644
--- a/lazuardy_anfis/__init__.py
+++ b/lazuardy_anfis/__init__.py
@@ -8,3 +8,7 @@
 - twmeggs <twmeggs@gmail.com>
 - lazuardy-tech <contact@lazuardy.tech>
 """
+
+from . import anfis, membershipfunction, mfderivs
+
+__all__ = ["anfis", "membershipfunction", "mfderivs"]
diff --git a/lazuardy_anfis/anfis.py b/lazuardy_anfis/anfis.py
index 33d8dbb..455c37c 100644
--- a/lazuardy_anfis/anfis.py
+++ b/lazuardy_anfis/anfis.py
@@ -14,7 +14,7 @@
 
 import numpy as np
 
-from lazuardy_anfis import mfderivs
+from . import mfderivs
 
 
 class ANFIS:
diff --git a/lazuardy_anfis/tests.py b/lazuardy_anfis/tests.py
index 329da49..8d468b4 100644
--- a/lazuardy_anfis/tests.py
+++ b/lazuardy_anfis/tests.py
@@ -11,11 +11,10 @@
 
 import os
 
-import anfis
-import membershipfunction
-import mfderivs
 import numpy as np
 
+from . import anfis, membershipfunction, mfderivs
+
 # define the training_set.txt file path
 training_set = os.getcwd() + "/lazuardy_anfis/training_set.txt"
 
diff --git a/setup.py b/setup.py
index e90e3bb..3cc325c 100644
--- a/setup.py
+++ b/setup.py
@@ -14,16 +14,7 @@
     long_description_content_type="text/markdown",
     author="Lazuardy",
     author_email="contact@lazuardy.tech",
-    packages=find_packages(
-        include=[
-            "anfis",
-            "fuzzy-logic",
-            "neural-networks",
-            "fnn",
-            "lazuardy",
-            "lazuardy_anfis",
-        ]
-    ),
+    packages=find_packages(),
     keywords="anfis, fuzzy logic, neural networks, fnn, lazuardy, lazuardy anfis",
     url="https://github.com/lazuardy-tech/anfis",
     download_url="https://github.com/lazuardy-tech/anfis/releases",