diff --git a/docs/index.md b/docs/index.md
index a6bcde60c0b5b8300c5af79cf9acb052593db52c..0bc2998870eab0fb4dc61f642db7c4b1db42efd4 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -4,15 +4,9 @@ This package provides you:
 - Preprocessing (orientation calculation, action calculation, raycasting, ...)
 - Quick and easy evaluation of behavior
 - Data, which is interchangable between labs and tools. No conversions required, since units and coordinate systems are standardized.
-- No custom data import, just `include robofish.io`
+- No custom data import, just `import robofish.io`
 - This package is tested extensively
 
-Features coming up:
-
-- Interface for unified behavior models
-- Pytorch Datasets directly from `robofish.io` files.
-
-
 <a href="https://git.imp.fu-berlin.de/bioroboticslab/robofish/io" class="myButton">Code</a>
 
 ## Installation
diff --git a/setup.py b/setup.py
index ccd90bde966f910f4c8a583d8a0663b2c44180a7..69899005b441bd7fdf26ccef89f9b44d0d8389f8 100644
--- a/setup.py
+++ b/setup.py
@@ -62,7 +62,6 @@ setup(
         "tqdm",
         "pre-commit",
         "scipy",
-        "torch",
     ],
     classifiers=[
         "Development Status :: 3 - Alpha",