diff --git a/bert/__init__.py b/bert/__init__.py
index d0e3123..2574243 100644
--- a/bert/__init__.py
+++ b/bert/__init__.py
@@ -8,4 +8,6 @@
 from .layer import Layer
 from .model import BertModelLayer
 
-from .loader import StockBertConfig, load_stock_weights
+from .loader import StockBertConfig, load_stock_weights, params_from_pretrained_ckpt
+from .tokenization import FullTokenizer
+