Cosmetics.

diff --git a/README.md b/README.md
index 3f347bb..d6e302c 100644 (file)
--- a/README.md
+++ b/README.md
@@ -6,12 +6,12 @@ the pytorch framework.
 The main function is
 
 ```
-torch.ByteTensor generate_sequence(bool pulling,
-                                   int nb_sequences,
-                                   int nb_images,
-                                   int image_height, int image_width,
-                                   int nb_shapes,
-                                   bool random_shape_size, bool random_colors)
+torch.ByteTensor flatland.generate_sequence(bool pulling,
+                                            int nb_sequences,
+                                            int nb_images,
+                                            int image_height, int image_width,
+                                            int nb_shapes,
+                                            bool random_shape_size, bool random_colors)
 ```
 
 where