diff --git a/dlclive/__init__.py b/dlclive/__init__.py
index 482ea6b..15d0731 100644
--- a/dlclive/__init__.py
+++ b/dlclive/__init__.py
@@ -14,5 +14,17 @@
 from dlclive.benchmark import benchmark_videos, download_benchmarking_data
 from dlclive.display import Display
 from dlclive.dlclive import DLCLive
+from dlclive.engine import Engine
 from dlclive.processor.processor import Processor
 from dlclive.version import VERSION, __version__
+
+__all__ = [
+    "DLCLive",
+    "Display",
+    "Processor",
+    "Engine",
+    "benchmark_videos",
+    "download_benchmarking_data",
+    "VERSION",
+    "__version__",
+]