ASEM000 · ASEM000 · Dec 19, 2023 · Dec 19, 2023 · Dec 19, 2023
diff --git a/docs/API/core.rst b/docs/API/core.rst
@@ -0,0 +1,31 @@
+🎯 Core API
+=============================
+
+
+.. currentmodule:: serket 
+
+.. autoclass:: TreeClass 
+    :members:
+        at
+.. autoclass:: Partial
+.. autoclass:: partial
+.. autoclass:: AtIndexer
+    :members:
+        get,
+        set,
+        apply,
+        scan,
+        reduce,
+        pluck,
+
+.. autoclass:: at
+.. autoclass:: BaseKey
+    :members:
+        __eq__
+.. autofunction:: autoinit
+.. autofunction:: leafwise
+.. autofunction:: field
+.. autofunction:: fields
+.. autofunction:: bcmap
+.. autofunction:: is_tree_equal
+.. autofunction:: value_and_tree
diff --git a/docs/API/masking.rst b/docs/API/masking.rst
@@ -0,0 +1,11 @@
+🥽 Masking API
+----------------------------------------------
+
+.. currentmodule:: serket 
+
+.. autofunction:: is_nondiff
+.. autofunction:: freeze
+.. autofunction:: unfreeze
+.. autofunction:: is_frozen
+.. autofunction:: tree_mask
+.. autofunction:: tree_unmask
diff --git a/docs/API/pretty_print.rst b/docs/API/pretty_print.rst
@@ -0,0 +1,11 @@
+🎨 Pretty printing API
+----------------------------------------------
+
+.. currentmodule:: serket 
+
+.. autofunction:: tree_diagram
+.. autofunction:: tree_graph
+.. autofunction:: tree_mermaid 
+.. autofunction:: tree_repr 
+.. autofunction:: tree_str
+.. autofunction:: tree_summary
diff --git a/docs/API/sepes.rst b/docs/API/sepes.rst
@@ -3,4 +3,12 @@
 
 .. note::
     `sepes <https://sepes.readthedocs.io/en/latest/?badge=latest>`_ API is fully re-exported under the ``serket`` namespace.
-    `Check the docs <https://sepes.readthedocs.io/en/latest/?badge=latest>`_ for full details.
+    `Check the docs <https://sepes.readthedocs.io/en/latest/?badge=latest>`_ for full details.
+
+.. toctree::
+   :maxdepth: 2
+   :caption: API Documentation
+
+   core
+   masking
+   pretty_print
diff --git a/docs/_static/tree_graph.svg b/docs/_static/tree_graph.svg
diff --git a/docs/_static/tree_graph_stylized.svg b/docs/_static/tree_graph_stylized.svg
diff --git a/docs/_static/tree_mermaid copy.jpg b/docs/_static/tree_mermaid copy.jpg
diff --git a/docs/notebooks/fields.ipynb b/docs/notebooks/fields.ipynb
@@ -22,7 +22,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "This section introduces common recipes for fields. A `field` is a type hinted class variable that adds certain functionality to the class. For example, a `field` can be used to validate the input data, or to provide a default value. The notebook provides examples for common use cases."
+    "This section introduces common recipes for fields. A `field` is a type hinted class variable that adds certain functionality to the class. For example, a `field` can be used to validate the input data, or to provide a default value. The notebook provides examples for common use cases. `serket.field` is implemented as a [python descriptor](https://docs.python.org/3/howto/descriptor.html), which means that it can be used in any class not necessarily a `serket` class. Refer to the [python documentation](https://docs.python.org/3/howto/descriptor.html) for more information on descriptors and how they work."
    ]
   },
   {