diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py
index 7e17b3940bdb7615703f786aafb565fe33db0984..f23ca73ec33489f80b17f1aa95a245cb30163173 100644
--- a/examples/Simple_Realistic_Railway_Generator.py
+++ b/examples/Simple_Realistic_Railway_Generator.py
@@ -316,10 +316,14 @@ def realistic_rail_generator(num_cities=5,
         return nodes_added
 
     def connect_stations(rail_trans, rail_array, org_s_nodes, org_e_nodes, nodes_added,
-                         inter_connect_max_nbr_of_shortes_city):
+                         inter_connect_max_nbr_of_shortes_city, start_to_end=True):
 
-        s_nodes = org_s_nodes.copy()
-        e_nodes = org_e_nodes.copy()
+        if start_to_end:
+            s_nodes = org_s_nodes.copy()
+            e_nodes = org_e_nodes.copy()
+        else:
+            e_nodes = org_s_nodes.copy()
+            s_nodes = org_e_nodes.copy()
 
         for city_loop in range(len(s_nodes)):
             old_cl = []
@@ -358,6 +362,9 @@ def realistic_rail_generator(num_cities=5,
                         rail_array[start_node] = tmp_trans_sn
                         rail_array[end_node] = tmp_trans_en
 
+        if start_to_end:
+            connect_stations(rail_trans, rail_array, org_s_nodes, org_e_nodes, nodes_added,
+                             inter_connect_max_nbr_of_shortes_city, start_to_end=False)
 
     def connect_random_stations(rail_trans, rail_array, start_nodes_added, end_nodes_added, nodes_added,
                                 inter_connect_max_nbr_of_shortes_city):
@@ -525,14 +532,14 @@ def realistic_rail_generator(num_cities=5,
 for itrials in range(100):
     print(itrials, "generate new city")
     np.random.seed(int(time.time()))
-    env = RailEnv(width=100,  # 20+np.random.choice(100),
-                  height=100,  # 20+np.random.choice(100),
+    env = RailEnv(width=30+np.random.choice(100),
+                  height=30+np.random.choice(100),
                   rail_generator=realistic_rail_generator(num_cities=np.random.choice(10) + 2,
                                                           city_size=np.random.choice(10) + 10,
                                                           allowed_rotation_angles=np.arange(0, 360, 90),
                                                           max_number_of_station_tracks=4,
                                                           nbr_of_switches_per_station_track=2,
-                                                          connect_max_nbr_of_shortes_city=4,
+                                                          connect_max_nbr_of_shortes_city=2,
                                                           do_random_connect_stations=False,
                                                           # Number of cities in map
                                                           seed=int(time.time())  # Random seed