diff --git a/baukit/__init__.py b/baukit/__init__.py
index 4245eef..a558172 100644
--- a/baukit/__init__.py
+++ b/baukit/__init__.py
@@ -8,9 +8,13 @@ from . import pbar
 from .nethook import Trace, TraceDict, set_requires_grad
 from .nethook import module_names, parameter_names
 from .nethook import subsequence, get_module, get_parameter, replace_module
-from .runningstats import Stat, Mean, Variance, Covariance, Bincount, CrossCovariance
-from .runningstats import IoU, CrossIoU, Quantile, TopK, History, CombinedStat
+from .pidfile import reserve_dir
+from .parallelfolder import ParallelImageFolders
+from .runningstats import Stat, Mean, Variance, Covariance, Bincount
+from .runningstats import CrossCovariance, IoU, CrossIoU, Quantile, TopK
+from .runningstats import Reservoir, History, CombinedStat
 from .runningstats import tally
 from . import show
 from .workerpool import WorkerBase, WorkerPool
-from .pidfile import reserve_dir
+from .tokendataset import TokenizedDataset, dict_to_, length_collation
+from .tokendataset import make_padded_batch, flatten_masked_batch