diff --git a/.github/pull_request_template.md b/.github/pull_request_template.md
index 94721bc75..d16bf21fe 100644
--- a/.github/pull_request_template.md
+++ b/.github/pull_request_template.md
@@ -6,14 +6,14 @@ This PR introduces:
 - [modification 1]
 - [modification 2]
 
-
 ## Checks
 
 - [ ] a clear description of the PR has been added
 - [ ] sufficient tests have been written
-- [ ] relevant section added to the documentation
-- [ ] example notebook added to the repo
+- [ ] example notebook have been added to the repo
+- [ ] all example notebooks have been verified to execute without errors
 - [ ] clean docstrings and comments have been written
+- [ ] relevant section have been added to the documentation
 - [ ] if any issue/observation has been discovered, a new issue has been opened
 
 ## Future improvements
diff --git a/qdax/__init__.py b/qdax/__init__.py
index 6a9beea82..3d26edf77 100644
--- a/qdax/__init__.py
+++ b/qdax/__init__.py
@@ -1 +1 @@
-__version__ = "0.4.0"
+__version__ = "0.4.1"
diff --git a/requirements.txt b/requirements.txt
index f6dea29a5..9422741a1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,12 +2,9 @@ absl-py==1.0.0
 brax==0.10.4
 chex==0.1.86
 flax==0.8.5
-gym==0.26.2
-ipython
 jax==0.4.28
 jaxlib==0.4.28
 jumanji==0.3.1
-jupyter
 numpy==1.26.4
 optax==0.1.9
 protobuf==3.19.5
diff --git a/setup.py b/setup.py
index cd7d2b131..8d651e102 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,6 @@
         "brax>=0.10.4",
         "chex>=0.1.86",
         "flax>=0.8.5",
-        "gym>=0.26.2",
         "jax>=0.4.28",
         "jaxlib>=0.4.28",  # necessary to build the doc atm
         "jinja2>=3.1.4",