diff --git a/docs/index.rst b/docs/index.rst
index ad84d9d..32cb4e9 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -10,4 +10,9 @@ TensorNEAT documentation
    :maxdepth: 2
    :caption: Contents:
 
-   source/tensorneat
\ No newline at end of file
+   source/tensorneat
+
+.. automodule:: tensorneat
+   :members:
+   :undoc-members:
+   :show-inheritance: