diff --git a/irm_kmi_api/__init__.py b/irm_kmi_api/__init__.py
index 95f0762..fa67793 100644
--- a/irm_kmi_api/__init__.py
+++ b/irm_kmi_api/__init__.py
@@ -1,10 +1,45 @@
-from .api import (IrmKmiApiClient, IrmKmiApiClientHa,
-                  IrmKmiApiCommunicationError, IrmKmiApiError)
-from .data import (AnimationFrameData, ConditionEvol, CurrentWeatherData,
-                   ExtendedForecast, Forecast, PollenLevel, PollenName,
-                   RadarAnimationData, RadarForecast, RadarStyle, WarningData,
-                   WarningType)
+from .api import (
+    IrmKmiApiClient,
+    IrmKmiApiClientHa,
+    IrmKmiApiCommunicationError,
+    IrmKmiApiError,
+)
+from .data import (
+    AnimationFrameData,
+    ConditionEvol,
+    CurrentWeatherData,
+    ExtendedForecast,
+    Forecast,
+    PollenLevel,
+    PollenName,
+    RadarAnimationData,
+    RadarForecast,
+    RadarStyle,
+    WarningData,
+    WarningType,
+)
 from .pollen import PollenParser
 from .rain_graph import RainGraph
 
+__all__ = [
+    "IrmKmiApiClient",
+    "IrmKmiApiClientHa",
+    "IrmKmiApiCommunicationError",
+    "IrmKmiApiError",
+    "AnimationFrameData",
+    "ConditionEvol",
+    "CurrentWeatherData",
+    "ExtendedForecast",
+    "Forecast",
+    "PollenLevel",
+    "PollenName",
+    "RadarAnimationData",
+    "RadarForecast",
+    "RadarStyle",
+    "WarningData",
+    "WarningType",
+    "PollenParser",
+    "RainGraph"
+]
+
 __version__ = '0.1.6'