diff --git a/examples/generate_data.py b/examples/generate_data.py
index fb15882..6cfbd3f 100644
--- a/examples/generate_data.py
+++ b/examples/generate_data.py
@@ -9,21 +9,32 @@ class Movie(BaseModel):
     title: str
     year: int
 
+
 class MovieCharecter(BaseModel):
     name: str
     actor: str
 
+
 class MovieQuote(BaseModel):
     quote: str
     movie: Movie
     charecter: MovieCharecter
 
+
 class QuotesList(BaseModel):
     quotes: List[MovieQuote]
     theme: str
 
 
-quotes = sm.generate_data(llm_provider="openai", llm_model="gpt-4o-mini", prompt="Generate 20 quotes from famous movies", response_model=QuotesList)
+quotes = sm.generate_data(
+    llm_provider="openai",
+    llm_model="gpt-4o-mini",
+    prompt="Generate 20 quotes from famous movies",
+    response_model=QuotesList,
+)
 
-for quote in quotes.quotes:
-    print(f"{quote.charecter.name} from {quote.movie.title} ({quote.movie.year}): {quote.quote!r}")
+if __name__ == "__main__":
+    for quote in quotes.quotes:
+        print(
+            f"{quote.charecter.name} from {quote.movie.title} ({quote.movie.year}): {quote.quote!r}"
+        )