diff --git a/examples/training_navigation.py b/examples/training_navigation.py
index dabeafee356f909b536700757182321043f7eada..b60468a51b8b73a4448a57494357328b130bb7b5 100644
--- a/examples/training_navigation.py
+++ b/examples/training_navigation.py
@@ -30,10 +30,10 @@ env = RailEnv(width=10,
               rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability),
               number_of_agents=1)
 """
-env = RailEnv(width=10,
-              height=10,
-              rail_generator=complex_rail_generator(nr_start_goal=3, min_dist=5, max_dist=99999, seed=0),
-              number_of_agents=3)
+env = RailEnv(width=50,
+              height=50,
+              rail_generator=complex_rail_generator(nr_start_goal=50, min_dist=5, max_dist=99999, seed=0),
+              number_of_agents=5)
 """
 env = RailEnv(width=20,
               height=20,
@@ -59,9 +59,9 @@ scores = []
 dones_list = []
 action_prob = [0] * 4
 agent = Agent(state_size, action_size, "FC", 0)
-#agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint1300.pth'))
+agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint1500.pth'))
 
-demo = False
+demo = True
 
 
 def max_lt(seq, val):
diff --git a/flatland/baselines/Nets/avoid_checkpoint15000.pth b/flatland/baselines/Nets/avoid_checkpoint15000.pth
index ca019b7b5d221577bcdb65e3979ba9795e5fd65b..adcfe61576553bbf0e2b4ba00d9fffafbfd9d7da 100644
Binary files a/flatland/baselines/Nets/avoid_checkpoint15000.pth and b/flatland/baselines/Nets/avoid_checkpoint15000.pth differ