diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
index 44fd6ae..d9b8a0c 100644
--- a/CONTRIBUTING.md
+++ b/CONTRIBUTING.md
@@ -38,7 +38,7 @@ To set up a local development environment:
    ```
 2. Install dependencies
    ```bash
-   pip install -r requirements.txt
+   pip install -r requirements_dev.txt
    ```
 
 ## Style Guide
diff --git a/requirements_dev.txt b/requirements_dev.txt
new file mode 100644
index 0000000..6ae0d11
--- /dev/null
+++ b/requirements_dev.txt
@@ -0,0 +1,3 @@
+-r requirements.txt  # Include production deps
+pytest
+black
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 2cc13e2..640e734 100644
--- a/setup.py
+++ b/setup.py
@@ -7,10 +7,11 @@
     name="tofupilot",
     version="1.9.4",
     packages=find_packages(),
-    install_requires=["requests", "setuptools", "packaging", "pytest", "websockets"],
-    entry_points={
-        "pytest11": [
-            "tofupilot = tofupilot.plugin",  # Registering the pytest plugin
+    install_requires=["requests", "setuptools", "packaging", "websockets"],
+    extras_require={
+        "dev": [
+            "pytest",
+            "black",
         ],
     },
     author="Félix Berthier",
diff --git a/tofupilot/__init__.py b/tofupilot/__init__.py
index 267a9a1..772180e 100644
--- a/tofupilot/__init__.py
+++ b/tofupilot/__init__.py
@@ -6,5 +6,4 @@
 """
 
 from .client import TofuPilotClient
-from .plugin import numeric_step, string_step, conf
 from .models import MeasurementOutcome, PhaseOutcome