diff --git a/lmm_tools/__init__.py b/lmm_tools/__init__.py
index 398534f6..7e7db515 100644
--- a/lmm_tools/__init__.py
+++ b/lmm_tools/__init__.py
@@ -1,3 +1,3 @@
-from .lmm import *
-from .emb import *
-from .data import *
+from .lmm import LMM, LLaVALMM, OpenAILMM, get_lmm
+from .emb import Embedder, SentenceTransformerEmb, OpenAIEmb, get_embedder
+from .data import DataStore, build_data_store