diff --git a/extras/fileformats/extras/datascience/__init__.py b/extras/fileformats/extras/datascience/__init__.py
index d94fbe5..efe7ad2 100644
--- a/extras/fileformats/extras/datascience/__init__.py
+++ b/extras/fileformats/extras/datascience/__init__.py
@@ -1,7 +1,7 @@
 from ._version import __version__
 from pathlib import Path
 import typing as ty
-from fileformats.core import extra_implementation
+from fileformats.core import extra_implementation, SampleFileGenerator
 from fileformats.datascience import TextMatrix
 
 __all__ = ["__version__"]
@@ -9,8 +9,9 @@
 
 @extra_implementation(TextMatrix.generate_sample_data)
 def text_matrix_generate_sample_data(
-    matrix: TextMatrix, dest_dir: Path
+    matrix: TextMatrix,
+    generator: SampleFileGenerator,
 ) -> ty.List[Path]:
-    fspath = dest_dir / "text-matrix.mat"
+    fspath = generator.dest_dir / "text-matrix.mat"
     fspath.write_text("0 1 2 3 4 5\n10 11 12 13 14 15\n20 21 22 23 24 25\n")
     return [fspath]
diff --git a/fileformats/datascience/__init__.py b/fileformats/datascience/__init__.py
index 9bac654..af22e17 100644
--- a/fileformats/datascience/__init__.py
+++ b/fileformats/datascience/__init__.py
@@ -1,6 +1,8 @@
 from ._version import __version__
 from .data import (
     TextMatrix,
+    TextArray,
+    TextVector,
     MatFile,
     RData,
     DatFile,
@@ -15,6 +17,8 @@
 __all__ = [
     "__version__",
     "TextMatrix",
+    "TextArray",
+    "TextVector",
     "MatFile",
     "RData",
     "DatFile",