diff --git a/langalf/agent.py b/langalf/agent.py
index 4320483..f2ea8fa 100644
--- a/langalf/agent.py
+++ b/langalf/agent.py
@@ -1,5 +1,4 @@
-"""
-Quality Assurance Testing Agent
+"""Quality Assurance Testing Agent.
 
 The goal of this agent is to perform quality assurance testing on a product or service.
 
diff --git a/langalf/http_spec.py b/langalf/http_spec.py
index 2c5d774..cbf12e8 100644
--- a/langalf/http_spec.py
+++ b/langalf/http_spec.py
@@ -1,4 +1,4 @@
-from typing import Any
+
 import httpx
 from pydantic import BaseModel
 
@@ -14,8 +14,7 @@ class LLMSpec(BaseModel):
         return parse_http_spec(http_spec)
 
     async def probe(self, prompt: str) -> httpx.Response:
-        """
-        Sends an HTTP request using the `httpx` library.
+        """Sends an HTTP request using the `httpx` library.
 
         Replaces a placeholder in the request body with a provided prompt and returns the response.
 
@@ -42,8 +41,7 @@ class LLMSpec(BaseModel):
 
 
 def parse_http_spec(http_spec: str) -> LLMSpec:
-    """
-    Parses an HTTP specification string into a LLMSpec object.
+    """Parses an HTTP specification string into a LLMSpec object.
 
     Args:
         http_spec (str): A string representing an HTTP specification.
@@ -79,7 +77,8 @@ def parse_http_spec(http_spec: str) -> LLMSpec:
 
 
 def escape_special_chars_for_json(prompt: str) -> str:
-    """Escapes special characters in a string for safe inclusion in a JSON template.
+    """Escapes special characters in a string for safe inclusion in a JSON
+    template.
 
     Args:
         prompt (str): The input string to be escaped.
diff --git a/langalf/probe_data/data.py b/langalf/probe_data/data.py
index 34fe518..da28f88 100644
--- a/langalf/probe_data/data.py
+++ b/langalf/probe_data/data.py
@@ -296,8 +296,8 @@ def load_local_csv() -> ProbeDataset:
 
 
 def dataset_from_iterator(name: str, iterator) -> list:
-    """
-    Convert an iterator into a list of prompts and create a ProbeDataset object.
+    """Convert an iterator into a list of prompts and create a ProbeDataset
+    object.
 
     Args:
         name (str): The name of the dataset.
@@ -305,7 +305,6 @@ def dataset_from_iterator(name: str, iterator) -> list:
 
     Returns:
         list: A list containing a single ProbeDataset object.
-
     """
     prompts = list(iterator)
     tokens = count_words_in_list(prompts)
diff --git a/langalf/probe_data/test_data.py b/langalf/probe_data/test_data.py
index 139714c..9a0cfda 100644
--- a/langalf/probe_data/test_data.py
+++ b/langalf/probe_data/test_data.py
@@ -1,5 +1,6 @@
 from inline_snapshot import snapshot
-from .data import prepare_prompts, ProbeDataset
+
+from .data import ProbeDataset, prepare_prompts
 
 
 class TestPreparePrompts: