diff --git a/docs/_templates/custom_module.rst b/docs/_templates/custom_module.rst
index 9eb45c5..466d2a8 100644
--- a/docs/_templates/custom_module.rst
+++ b/docs/_templates/custom_module.rst
@@ -1,8 +1,8 @@
 {{ objname | escape | underline}}
 
 .. automodule:: {{ fullname }}
-    :members:
-    :undoc-members:
+   :members:
+   :undoc-members:
 
    {% block attributes %}
    {% if attributes %}
diff --git a/docs/_templates/custom_summary.rst b/docs/_templates/custom_summary.rst
index 6689efc..f579cd9 100644
--- a/docs/_templates/custom_summary.rst
+++ b/docs/_templates/custom_summary.rst
@@ -1,8 +1,8 @@
 {{ fullname | escape | underline}}
 
 .. automodule:: {{ fullname }}
-    :members:
-    :undoc-members:
+   :members:
+   :undoc-members:
 
    {% block attributes %}
    {% if attributes %}
diff --git a/docs/conf.py b/docs/conf.py
index 53432fd..47d9edc 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -62,7 +62,7 @@
 autodoc_mock_imports = []
 
 # Controls how to represent typehints.
-autodoc_typehints = "description"
+autodoc_typehints = "signature"
 
 # -- Options for HTML output -------------------------------------------------
 
diff --git a/src/arcade_collection/input/__init__.py b/src/arcade_collection/input/__init__.py
index e588637..246f5d3 100644
--- a/src/arcade_collection/input/__init__.py
+++ b/src/arcade_collection/input/__init__.py
@@ -1,3 +1,5 @@
+"""Tasks for generating input files for ARCADE."""
+
 from prefect import task
 
 from .convert_to_cells_file import convert_to_cells_file
diff --git a/src/arcade_collection/output/__init__.py b/src/arcade_collection/output/__init__.py
index 82b9470..1dff9a6 100644
--- a/src/arcade_collection/output/__init__.py
+++ b/src/arcade_collection/output/__init__.py
@@ -1,3 +1,5 @@
+"""Tasks for processing output files from ARCADE."""
+
 from prefect import task
 
 from .convert_model_units import convert_model_units