diff --git a/docs/conf.py b/docs/conf.py
index 61ed9b2..422bd67 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -54,9 +54,13 @@
 autoapi_ignore = ["*/__main__.py", "*/_version.py"]
 autoapi_add_toc_tree_entry = False
 autoapi_member_order = "bysource"
-autoapi_options = ['members', 'undoc-members', 'private-members',
-                   'show-inheritance', 'show-module-summary',
-                   'special-members',]
+autoapi_options = [
+    "members",
+    "undoc-members",
+    "show-inheritance",
+    "show-module-summary",
+    "special-members",
+]
 
 html_theme = "sphinx_book_theme"
 
diff --git a/src/nested_pandas/datasets/__init__.py b/src/nested_pandas/datasets/__init__.py
index d90b287..4b8e827 100644
--- a/src/nested_pandas/datasets/__init__.py
+++ b/src/nested_pandas/datasets/__init__.py
@@ -1 +1 @@
-from .generation import generate_data, generate_parquet_file  # noqa
+from .generation import *  # noqa
diff --git a/src/nested_pandas/example_benchmarks.py b/src/nested_pandas/example_benchmarks.py
deleted file mode 100644
index 5a77b06..0000000
--- a/src/nested_pandas/example_benchmarks.py
+++ /dev/null
@@ -1,14 +0,0 @@
-"""An example module containing simplistic methods under benchmarking."""
-
-import random
-import time
-
-
-def runtime_computation():
-    """Runtime computation consuming between 0 and 5 seconds."""
-    time.sleep(random.uniform(0, 5))
-
-
-def memory_computation():
-    """Memory computation for a random list up to 512 samples."""
-    return [0] * random.randint(0, 512)