diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index 3ca6ab9d342c0b3a2907252264274899f8e519f7..621f2c797be255cf8530d5d079932ed5daee0db4 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -30,14 +30,14 @@ speed_ration_map = {1.: 0.25,  # Fast passenger train
 
 env = RailEnv(width=50,
               height=50,
-              rail_generator=sparse_rail_generator(num_cities=2,  # Number of cities in map (where train stations are)
+              rail_generator=sparse_rail_generator(num_cities=8,  # Number of cities in map (where train stations are)
                                                    seed=1,  # Random seed
                                                    grid_mode=False,
                                                    max_inter_city_rails=2,
                                                    max_tracks_in_city=4,
                                                    ),
               schedule_generator=sparse_schedule_generator(),
-              number_of_agents=50,
+              number_of_agents=10,
               stochastic_data=stochastic_data,  # Malfunction data generator
               obs_builder_object=GlobalObsForRailEnv())
 
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index a1205580d89df199fc23b0f9c038764b3984044c..337b301a1c9005adaff311dd442310386d287b21 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -946,7 +946,6 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
 
             if direction_set == 4:
                 return closest_neighb
-        print(closest_neighb)
         return closest_neighb
 
     def argsort(seq):