diff --git a/generator/simple/simple_generator.py b/generator/simple/simple_generator.py
index 0a4d78c..6db0667 100644
--- a/generator/simple/simple_generator.py
+++ b/generator/simple/simple_generator.py
@@ -18,7 +18,7 @@ tf.logging.set_verbosity(tf.logging.ERROR)
 
 class SimpleGenerator:
 
-    def __init__(self,  generate_num=40, temperature=0.9, top_k=40, top_p=0.0):
+    def __init__(self,  generate_num=65, temperature=0.9, top_k=40, top_p=0.0):
         self.generate_num=generate_num
         self.temp = temperature
         self.top_k = top_k
diff --git a/story/story_manager.py b/story/story_manager.py
index db60ae9..0381c64 100644
--- a/story/story_manager.py
+++ b/story/story_manager.py
@@ -41,7 +41,13 @@ class Story():
         self.results.append(story_block)
 
     def latest_result(self):
-        if len(self.results) >= 2:
+        if len(self.results) >= 4:
+            return self.context + self.actions[-4] + self.results[-4] + self.actions[-3] + self.results[-3] + \
+                   self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
+        elif len(self.results) >= 3:
+            return self.context + self.actions[-3] + self.results[-3] + self.actions[-2] + self.results[-2] + \
+                   self.actions[-1] + self.results[-1]
+        elif len(self.results) >= 2:
             return self.context + self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
         elif len(self.results) >= 1:
             return self.context + self.actions[-1] + self.results[-1]