######################################################################
 
 if __name__ == "__main__":
-    height, width, T = 4, 6, 20
-    states, actions, rewards = generate_episodes(3, height, width, T)
+    nb, height, width, T = 8, 4, 6, 20
+    states, actions, rewards = generate_episodes(nb, height, width, T)
     seq = episodes2seq(states, actions, rewards)
     s, a, r = seq2episodes(seq, height, width)
     print(episodes2str(s, a, r, unicode=True, ansi_colors=True))