diff --git a/docs/source/conf.py b/docs/source/conf.py
index eccdb5f1..0db6a74a 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -1,16 +1,15 @@
 import datetime
-import sphinx_rtd_theme
 import doctest
+
+import sphinx_rtd_theme
 import torch_scatter
 
 extensions = [
     'sphinx.ext.autodoc',
-    'sphinx.ext.doctest',
     'sphinx.ext.intersphinx',
     'sphinx.ext.mathjax',
     'sphinx.ext.napoleon',
     'sphinx.ext.viewcode',
-    'sphinx.ext.githubpages',
 ]
 
 source_suffix = '.rst'
diff --git a/docs/source/functions/composite.rst b/docs/source/functions/composite.rst
new file mode 100644
index 00000000..2aca0605
--- /dev/null
+++ b/docs/source/functions/composite.rst
@@ -0,0 +1,8 @@
+composite
+=========
+
+.. currentmodule:: torch_scatter.composite
+
+.. automodule:: torch_scatter.composite
+   :members:
+   :undoc-members:
diff --git a/docs/source/index.rst b/docs/source/index.rst
index a84a77d8..3481d4a9 100644
--- a/docs/source/index.rst
+++ b/docs/source/index.rst
@@ -14,7 +14,10 @@ All included operations are broadcastable, work on varying data types, are imple
    :maxdepth: 0
    :caption: Package reference
 
-   functions/*
+   functions/scatter
+   functions/segment_coo
+   functions/segment_csr
+   functions/composite
 
 Indices and tables
 ==================