diff --git a/fedn/fedn/common/tracer/mongotracer.py b/fedn/fedn/common/tracer/mongotracer.py
index 55711809b..0a3e28cdc 100644
--- a/fedn/fedn/common/tracer/mongotracer.py
+++ b/fedn/fedn/common/tracer/mongotracer.py
@@ -86,6 +86,13 @@ def set_round_data(self, round_data):
             '$push': {'reducer': round_data}}, True)
 
     def update_client_status(self, client_name, status):
+        """ Update client status in statestore.
+        :param client_name: The client name
+        :type client_name: str
+        :param status: The client status
+        :type status: str
+        :return: None
+        """
         datetime_now = datetime.now()
         filter_query = {"name": client_name}
 
diff --git a/fedn/fedn/network/statestore/mongostatestore.py b/fedn/fedn/network/statestore/mongostatestore.py
index 0047615e4..19d514f59 100644
--- a/fedn/fedn/network/statestore/mongostatestore.py
+++ b/fedn/fedn/network/statestore/mongostatestore.py
@@ -114,8 +114,15 @@ def transition(self, state):
     def get_sessions(self, limit=None, skip=None, sort_key="_id", sort_order=pymongo.DESCENDING):
         """Get all sessions.
 
-        :return: All sessions.
-        :rtype: ObjectID
+        :param limit: The maximum number of sessions to return.
+        :type limit: int
+        :param skip: The number of sessions to skip.
+        :type skip: int
+        :param sort_key: The key to sort by.
+        :type sort_key: str
+        :param sort_order: The sort order.
+        :type sort_order: pymongo.ASCENDING or pymongo.DESCENDING
+        :return: Dictionary of sessions in result (array of session objects) and count.
         """
 
         result = None
@@ -616,11 +623,16 @@ def get_client(self, name):
         except Exception:
             return None
 
-    def list_clients(self, limit=None, skip=None, status=False, sort_key="last_seen", sort_order=pymongo.DESCENDING):
+    def list_clients(self, limit=None, skip=None, status=None, sort_key="last_seen", sort_order=pymongo.DESCENDING):
         """List all clients registered on the network.
 
-        :return: list of clients.
-        :rtype: list(ObjectId)
+        :param limit: The maximum number of clients to return.
+        :type limit: int
+        :param skip: The number of clients to skip.
+        :type skip: int
+        :param status:  online | offline
+        :type status: str
+        :param sort_key: The key to sort by.
         """
 
         result = None