diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 028928aeec2c08c7b9035d875a874987dce79896..3e90128c74bda860d8a4c75d71652071660482a3 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -343,6 +343,17 @@ def random_rail_generator(cell_type_relative_proportion=[1.0] * 11, seed=1) -> R
                 template = [template[-1]] + template[:-1]
 
         def get_matching_templates(template):
+            """
+            Returns a list of possible transition maps for a given template
+            
+            Parameters:
+            ------
+            template:List[int]
+
+            Returns:
+            ------
+            List[int]
+            """
             ret = []
             for i in range(len(transitions_templates_)):
                 is_match = True