diff --git a/generator/simple/simple_generator.py b/generator/simple/simple_generator.py
index 2c90172..0a4d78c 100644
--- a/generator/simple/simple_generator.py
+++ b/generator/simple/simple_generator.py
@@ -18,14 +18,14 @@ tf.logging.set_verbosity(tf.logging.ERROR)
 
 class SimpleGenerator:
 
-    def __init__(self,  generate_num=28, temperature=0.8, top_k=40, top_p=0.0):
+    def __init__(self,  generate_num=40, temperature=0.9, top_k=40, top_p=0.0):
         self.generate_num=generate_num
         self.temp = temperature
         self.top_k = top_k
         self.top_p = top_p
 
-        self.model_name = "1558M"
-        self.model_dir = "generator/simple/models"
+        self.model_name = "run1"
+        self.model_dir = "generator/simple/checkpoint"
         self.checkpoint_path = os.path.join(self.model_dir, self.model_name)
         if not os.path.isdir(os.path.join(self.model_dir, self.model_name)):
             print(f"Downloading {self.model_name} model...")
diff --git a/story/story_data.yaml b/story/story_data.yaml
index 584375b..868d11c 100644
--- a/story/story_data.yaml
+++ b/story/story_data.yaml
@@ -31,4 +31,4 @@ contexts:
 
   apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind. "
 
-  classic: "You enter a dungeon with your trusty sword and shield. You are searching for the evil necromancer who killed your family. You've heard that he resides at the bottom of the dungeon, guarded by legions of the undead. "
\ No newline at end of file
+  classic: "You are a knight in a deep dungeon. "
\ No newline at end of file