diff --git a/.gitignore b/.gitignore
index f1d17df..e18c66b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,9 +1,6 @@
 # .gitignore
 *
-!env_manager.py
+!env_manager/
 !README.md
 !setup.py
-!.gitignore
-!__init__.py
-!/env_manager/*
-!/env_manager
\ No newline at end of file
+!.gitignore
\ No newline at end of file
diff --git a/env_manager/__init__.py b/env_manager/__init__.py
index 024ad02..5b3cd41 100644
--- a/env_manager/__init__.py
+++ b/env_manager/__init__.py
@@ -1,3 +1,5 @@
-from env_manager import get_env
+from .env_manager import get_env
+
+# Initialize environment variables
 get_env()
 print("Environment variables set from ArangoDB by env_manager.")
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 25d1609..e99ff7b 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@ setup(
     ],
     entry_points={
         'console_scripts': [
-            'env_manager=env_manager:main',
+            'env_manager=env_manager.env_manager:main',
         ],
     },
 )
\ No newline at end of file