diff --git a/README.md b/README.md
index 9f1be80..baa75b2 100644
--- a/README.md
+++ b/README.md
@@ -12,7 +12,7 @@ In other words, within multi-modal motion planning scope, `mpot` enables better
 
 ## Installation
 
-Simply activate your conda/Python environment and run
+Simply activate your conda/Python environment, navigate to `mpot` root directory and run
 
 ```azure
 pip install -e .
diff --git a/requirements.txt b/requirements.txt
index 8ea60ff..341ca84 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
 scipy
 numpy
 matplotlib
+labmaze
 torch 
 einops
-torch_robotics @ git+https://github.com/anindex/torch_robotics@main
+torch_robotics @ git+https://github.com/anindex/torch_robotics.git@68d28399ff67f34ca3be0130acdc7cba2fabd61b
diff --git a/setup.py b/setup.py
index 77126c0..348a563 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,4 @@
-from setuptools import setup
+import setuptools
 from codecs import open
 from os import path
 
@@ -11,10 +11,11 @@
     for line in f:
         requires_list.append(str(line))
 
-setup(name='mpot',
-      description="Implementation of MPOT in PyTorch",
-      author="An T. Le",
-      author_email="an@robot-learning.de",
-      packages=['mpot'],
-      install_requires=requires_list,
+setuptools.setup(
+    name='mpot',
+    description="Implementation of MPOT in PyTorch",
+    author="An T. Le",
+    author_email="an@robot-learning.de",
+    packages=setuptools.find_namespace_packages(),
+    install_requires=requires_list,
 )