From 1ca53d128eafa9edccc3a68f96a8833b45e0686e Mon Sep 17 00:00:00 2001
From: Marvin Pfoertner <marvin.pfoertner@icloud.com>
Date: Fri, 28 Aug 2020 17:12:14 +0200
Subject: [PATCH] Further doc fixes

---
 docs/source/index.rst                    |  2 +-
 src/probnum/random_variables/__init__.py | 19 ++++++++++++++++---
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/docs/source/index.rst b/docs/source/index.rst
index c8c4a9f980..ee7a5fae48 100644
--- a/docs/source/index.rst
+++ b/docs/source/index.rst
@@ -48,7 +48,7 @@ To learn how to use ProbNum check out the `quickstart guide <introduction/quicks
    :maxdepth: 1
    :caption: API Documentation
 
-   public_api/prob
+   public_api/probnum
    public_api/linalg
    public_api/quad
    public_api/diffeq
diff --git a/src/probnum/random_variables/__init__.py b/src/probnum/random_variables/__init__.py
index 1264cc7e2b..cab71d5a24 100644
--- a/src/probnum/random_variables/__init__.py
+++ b/src/probnum/random_variables/__init__.py
@@ -16,17 +16,30 @@
     ContinuousRandomVariable,
 )
 
+from ._dirac import Dirac
+from ._normal import Normal
+
 from ._scipy_stats import (
     WrappedSciPyRandomVariable,
     WrappedSciPyDiscreteRandomVariable,
     WrappedSciPyContinuousRandomVariable,
 )
 
-from ._dirac import Dirac
-from ._normal import Normal
-
 from ._utils import asrandvar
 
+# Public classes and functions. Order is reflected in documentation.
+__all__ = [
+    "asrandvar",
+    "RandomVariable",
+    "DiscreteRandomVariable",
+    "ContinuousRandomVariable",
+    "Dirac",
+    "Normal",
+    "WrappedSciPyRandomVariable",
+    "WrappedSciPyDiscreteRandomVariable",
+    "WrappedSciPyContinuousRandomVariable",
+]
+
 # Set correct module paths. Corrects links and module paths in documentation.
 RandomVariable.__module__ = "probnum.random_variables"
 DiscreteRandomVariable.__module__ = "probnum.random_variables"