diff --git a/pyproject.toml b/pyproject.toml
index 9e81bf20..6e3cdc7f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "4.0.4"
+version = "4.0.5"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
diff --git a/swarms/telemetry/posthog_utils.py b/swarms/telemetry/posthog_utils.py
index 66bd0e6c..b3b2a73e 100644
--- a/swarms/telemetry/posthog_utils.py
+++ b/swarms/telemetry/posthog_utils.py
@@ -11,6 +11,16 @@ logger = logging.getLogger(__name__)
 
 
 class PosthogWrapper:
+    """
+    A wrapper class for interacting with the PostHog analytics service.
+
+    Args:
+        api_key (str): The API key for accessing the PostHog instance.
+        instance_address (str): The address of the PostHog instance.
+        debug (bool, optional): Whether to enable debug mode. Defaults to False.
+        disabled (bool, optional): Whether to disable tracking. Defaults to False.
+    """
+
     def __init__(
         self, api_key, instance_address, debug=False, disabled=False
     ):
@@ -19,9 +29,24 @@ class PosthogWrapper:
         self.posthog.disabled = disabled
 
     def capture_event(self, distinct_id, event_name, properties=None):
+        """
+        Capture an event in PostHog.
+
+        Args:
+            distinct_id (str): The distinct ID of the user.
+            event_name (str): The name of the event.
+            properties (dict, optional): Additional properties associated with the event. Defaults to None.
+        """
         self.posthog.capture(distinct_id, event_name, properties)
 
     def capture_pageview(self, distinct_id, url):
+        """
+        Capture a pageview event in PostHog.
+
+        Args:
+            distinct_id (str): The distinct ID of the user.
+            url (str): The URL of the page.
+        """
         self.posthog.capture(
             distinct_id, "$pageview", {"$current_url": url}
         )
@@ -29,6 +54,14 @@ class PosthogWrapper:
     def set_user_properties(
         self, distinct_id, event_name, properties
     ):
+        """
+        Set user properties in PostHog.
+
+        Args:
+            distinct_id (str): The distinct ID of the user.
+            event_name (str): The name of the event.
+            properties (dict): The user properties to set.
+        """
         self.posthog.capture(
             distinct_id, event=event_name, properties=properties
         )
@@ -36,21 +69,62 @@ class PosthogWrapper:
     def is_feature_enabled(
         self, flag_key, distinct_id, send_feature_flag_events=True
     ):
+        """
+        Check if a feature flag is enabled for a user.
+
+        Args:
+            flag_key (str): The key of the feature flag.
+            distinct_id (str): The distinct ID of the user.
+            send_feature_flag_events (bool, optional): Whether to send feature flag events. Defaults to True.
+
+        Returns:
+            bool: True if the feature flag is enabled, False otherwise.
+        """
         return self.posthog.feature_enabled(
             flag_key, distinct_id, send_feature_flag_events
         )
 
     def get_feature_flag_payload(self, flag_key, distinct_id):
+        """
+        Get the payload of a feature flag for a user.
+
+        Args:
+            flag_key (str): The key of the feature flag.
+            distinct_id (str): The distinct ID of the user.
+
+        Returns:
+            dict: The payload of the feature flag.
+        """
         return self.posthog.get_feature_flag_payload(
             flag_key, distinct_id
         )
 
     def get_feature_flag(self, flag_key, distinct_id):
+        """
+        Get the value of a feature flag for a user.
+
+        Args:
+            flag_key (str): The key of the feature flag.
+            distinct_id (str): The distinct ID of the user.
+
+        Returns:
+            str: The value of the feature flag.
+        """
         return self.posthog.get_feature_flag(flag_key, distinct_id)
 
     def capture_with_feature_flag(
         self, distinct_id, event_name, flag_key, variant_key
     ):
+        """
+        Capture an event with a feature flag in PostHog.
+
+        Args:
+            distinct_id (str): The distinct ID of the user.
+            event_name (str): The name of the event.
+            flag_key (str): The key of the feature flag.
+            variant_key (str): The key of the variant.
+
+        """
         self.posthog.capture(
             distinct_id,
             event_name,
@@ -60,6 +134,14 @@ class PosthogWrapper:
     def capture_with_feature_flags(
         self, distinct_id, event_name, send_feature_flags=True
     ):
+        """
+        Capture an event with all feature flags in PostHog.
+
+        Args:
+            distinct_id (str): The distinct ID of the user.
+            event_name (str): The name of the event.
+            send_feature_flags (bool, optional): Whether to send feature flags. Defaults to True.
+        """
         self.posthog.capture(
             distinct_id,
             event_name,