diff --git a/.bumpversion.cfg b/.bumpversion.cfg
index b3b3da2..9c7133a 100644
--- a/.bumpversion.cfg
+++ b/.bumpversion.cfg
@@ -6,7 +6,7 @@ parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)
 serialize = {major}.{minor}.{patch}
 message = build: bump version {current_version} → {new_version}
 
-[bumpversion:file:setup.py]
+[bumpversion:file:pyproject.toml]
 
 [bumpversion:file:./prototorch/models/__init__.py]
 
diff --git a/pyproject.toml b/pyproject.toml
index 0dec2e1..cca8884 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,9 +4,10 @@ name = "prototorch-models"
 version = "0.5.4"
 description = "Pre-packaged prototype-based machine learning models using ProtoTorch and PyTorch-Lightning."
 authors = [
+    { name = "Jensun Ravichandran", email = "jjensun@gmail.com" },
     { name = "Alexander Engelsberger", email = "engelsbe@hs-mittweida.de" },
 ]
-dependencies = ["lightning>=2.0.0", "prototorch>=0.7.5", "torchmetrics"]
+dependencies = ["lightning>=2.0.0", "prototorch>=0.7.5"]
 requires-python = ">=3.8"
 readme = "README.md"
 license = { text = "MIT" }
@@ -36,24 +37,31 @@ Downloads = "https://github.com/si-cim/prototorch_models.git"
 [project.optional-dependencies]
 dev = ["bumpversion", "pre-commit", "yapf", "toml"]
 examples = ["matplotlib", "scikit-learn"]
-tests = ["codecov", "pytest"]
-all = [
-    "bumpversion",
-    "codecov",
-    "ipykernel",
-    "jsonargparse",
-    "matplotlib",
-    "nbsphinx",
-    "pre-commit",
-    "yapf",
-    "toml",
-    "pytest",
+ci = ["pytest", "pre-commit"]
+docs = [
     "recommonmark",
-    "scikit-learn",
+    "nbsphinx",
     "sphinx",
     "sphinx_rtd_theme",
     "sphinxcontrib-bibtex",
     "sphinxcontrib-katex",
+    "ipykernel",
+]
+all = [
+    "bumpversion",
+    "pre-commit",
+    "yapf",
+    "toml",
+    "pytest",
+    "matplotlib",
+    "scikit-learn",
+    "recommonmark",
+    "nbsphinx",
+    "sphinx",
+    "sphinx_rtd_theme",
+    "sphinxcontrib-bibtex",
+    "sphinxcontrib-katex",
+    "ipykernel",
 ]
 
 [project.entry-points."prototorch.plugins"]