diff --git a/podcastfy/conversation_config.yaml b/podcastfy/conversation_config.yaml
index bbee6c5..904a67a 100644
--- a/podcastfy/conversation_config.yaml
+++ b/podcastfy/conversation_config.yaml
@@ -17,7 +17,7 @@ engagement_techniques:
   - "anecdotes"
   - "analogies"
   - "humor"
-creativity: 0
+creativity: 1
 
 text_to_speech:
   default_tts_model: "openai"
diff --git a/tests/test_genai_podcast.py b/tests/test_genai_podcast.py
index 0f56968..2cf5e76 100644
--- a/tests/test_genai_podcast.py
+++ b/tests/test_genai_podcast.py
@@ -11,16 +11,11 @@
 # TODO: Should be a fixture
 def sample_conversation_config():
     conversation_config = {
-        "word_count": 2000,
-        "conversation_style": ["formal", "educational"],
+        "word_count": 500,
         "roles_person1": "professor",
         "roles_person2": "student",
-        "dialogue_structure": ["Introduction", "Main Points", "Conclusion"],
         "podcast_name": "Teachfy",
-        "podcast_tagline": "Learning Through Conversation",
-        "output_language": "English",
-        "engagement_techniques": ["examples", "questions", "case studies"],
-        "creativity": 0,
+        "podcast_tagline": "Learning Through Conversation"
     }
     return conversation_config
 
@@ -54,7 +49,7 @@ def test_custom_conversation_config(self):
         """
         conversation_config = sample_conversation_config()
         content_generator = ContentGenerator(self.api_key, conversation_config)
-        input_text = "Artificial Intelligence in Education"
+        input_text = "United States of America"
 
         result = content_generator.generate_qa_content(input_text)
 
diff --git a/usage/config_custom.md b/usage/config_custom.md
index efe904b..7959a27 100644
--- a/usage/config_custom.md
+++ b/usage/config_custom.md
@@ -23,7 +23,7 @@ See [conversation_custom.md](conversation_custom.md) for more details.
 - `max_output_tokens`: 8192
   - Maximum number of tokens for the output generated by the AI model.
 - `temperature`: 0
-  - Controls randomness in the AI's output. 0 means deterministic responses.
+  - Controls randomness in the AI's output. 0 means deterministic responses. Range for gemini-1.5-pro: 0.0 - 2.0 (default: 1.0)
 - `langchain_tracing_v2`: true
   - Enables LangChain tracing for debugging and monitoring.
 
diff --git a/usage/conversation_custom.md b/usage/conversation_custom.md
index fbc4433..30a0594 100644
--- a/usage/conversation_custom.md
+++ b/usage/conversation_custom.md
@@ -146,7 +146,7 @@ custom_config = {
     "word_count": 200,
     "conversation_style": ["casual", "humorous"],
     "podcast_name": "Tech Chuckles",
-    "creativity": 7
+    "creativity": 0.7
 }
 
 generate_podcast(