diff --git a/custom_components/afvalwijzer/__init__.py b/custom_components/afvalwijzer/__init__.py
index 52900b8..496b7de 100644
--- a/custom_components/afvalwijzer/__init__.py
+++ b/custom_components/afvalwijzer/__init__.py
@@ -1,31 +1,31 @@
-# from homeassistant.config_entries import ConfigEntry
-# from homeassistant.core import HomeAssistant
-# from homeassistant.helpers.typing import ConfigType
-# from .const.const import DOMAIN
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.typing import ConfigType
+from .const.const import DOMAIN
 
 
-# async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
-#     """Set up the Afvalwijzer integration."""
-#     hass.data.setdefault(DOMAIN, {})
-#     return True
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
+    """Set up the Afvalwijzer integration."""
+    hass.data.setdefault(DOMAIN, {})
+    return True
 
 
-# async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
-#     """Set up Afvalwijzer from a config entry."""
-#     # Store config entry data
-#     hass.data[DOMAIN][entry.entry_id] = entry.data
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Set up Afvalwijzer from a config entry."""
+    # Store config entry data
+    hass.data[DOMAIN][entry.entry_id] = entry.data
 
-#     # Forward the setup to the sensor platform
-#     await hass.config_entries.async_forward_entry_setups(entry, ["sensor"])
-#     return True
+    # Forward the setup to the sensor platform
+    await hass.config_entries.async_forward_entry_setups(entry, ["sensor"])
+    return True
 
 
-# async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
-#     """Unload a config entry."""
-#     # Remove stored data
-#     if entry.entry_id in hass.data[DOMAIN]:
-#         hass.data[DOMAIN].pop(entry.entry_id)
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Unload a config entry."""
+    # Remove stored data
+    if entry.entry_id in hass.data[DOMAIN]:
+        hass.data[DOMAIN].pop(entry.entry_id)
 
-#     # Unload the sensor platform
-#     await hass.config_entries.async_forward_entry_unload(entry, "sensor")
-#     return True
+    # Unload the sensor platform
+    await hass.config_entries.async_forward_entry_unload(entry, "sensor")
+    return True
diff --git a/custom_components/afvalwijzer/collector/mijnafvalwijzer.py b/custom_components/afvalwijzer/collector/mijnafvalwijzer.py
index 2876e83..e1e21e3 100644
--- a/custom_components/afvalwijzer/collector/mijnafvalwijzer.py
+++ b/custom_components/afvalwijzer/collector/mijnafvalwijzer.py
@@ -25,16 +25,13 @@ def get_waste_data_raw(provider, postal_code, street_number, suffix):
 
     try:
         response = raw_response.json()
-        print(response)
         ophaaldagen_data = response.get("ophaaldagen", {}).get("data", [])
         ophaaldagen_next_data = response.get("ophaaldagenNext", {}).get("data", [])
 
-        # print(ophaaldagen_next_data)
-
         if not ophaaldagen_data and not ophaaldagen_next_data:
             _LOGGER.error("Address not found or no data available!")
             raise KeyError
     except KeyError as err:
         raise KeyError(f"Invalid and/or no data received from {url}") from err
 
-    return ophaaldagen_data + ophaaldagen_next_data
+    return ophaaldagen_data + ophaaldagen_next_data[:25]