diff --git a/src/__init__.py b/src/__init__.py
index e00194a..fa780ca 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -1 +1,6 @@
 # Infrared Fault Detection Package
+from .diagnose import FaultDiagnoser
+from .detect import InfraredDetector
+from .extract_temp import TempFeatureExtractor
+
+__all__ = ['FaultDiagnoser', 'InfraredDetector', 'TempFeatureExtractor']