diff --git a/playground/demos/ad_gen/ad_gen_example.py b/playground/demos/ad_gen/ad_gen_example.py
index 9cb2ca4a..37e0c18b 100644
--- a/playground/demos/ad_gen/ad_gen_example.py
+++ b/playground/demos/ad_gen/ad_gen_example.py
@@ -42,7 +42,7 @@ class ProductAdConceptGenerator:
             "serene",
             "lasers,lightning",
         ]
-        self.contexts = [
+        self.style = [
             "in an everyday setting",
             "in a rave setting",
             "in an abstract environment",
@@ -63,10 +63,21 @@ class ProductAdConceptGenerator:
         self.contexts = [
             "high realism product ad (extremely creative)"
         ]
+        self.social_media_platforms = [
+            "Facebook",
+            "Instagram",
+            "Twitter",
+            "LinkedIn",
+            "Snapchat",
+            "TikTok",
+            "Pinterest",
+            "YouTube",
+        ]
 
     def generate_concept(self):
         theme = random.choice(self.themes)
         context = random.choice(self.contexts)
+        style = random.choice(self.style)
         return (
             f"{theme} inside a {style} {self.product_name}, {context}"
         )
@@ -94,8 +105,7 @@ ad_copy_prompt = (
 ad_copy = ad_copy_agent.run(task=ad_copy_prompt)
 
 # Output the results
-print("Creative Concept:", concept_result)
-print("Design Ideas:", design_result)
+print("Creative Concept:", concept_concept)
 print("Ad Copy:", copywriting_result)
 print(
     "Image Path:",