diff --git a/exponax/__init__.py b/exponax/__init__.py
index 788197c..5685356 100644
--- a/exponax/__init__.py
+++ b/exponax/__init__.py
@@ -1,5 +1,6 @@
 from . import _metrics as metrics
 from . import _poisson as poisson
+from . import etdrk, ic, nonlin_fun, normalized, reaction, viz
 from ._base_stepper import BaseStepper
 from ._forced_stepper import ForcedStepper
 from ._repeated_stepper import RepeatedStepper
@@ -29,4 +30,10 @@
     "build_ic_set",
     "wrap_bc",
     "metrics",
+    "etdrk",
+    "ic",
+    "nonlin_fun",
+    "normalized",
+    "reaction",
+    "viz",
 ]