diff --git a/display/dislpay_2d.py b/display/dislpay_2d.py
index d6ef76e..747af87 100644
--- a/display/dislpay_2d.py
+++ b/display/dislpay_2d.py
@@ -37,7 +37,7 @@ class Displayer:
 
 
     def register_object(self, obj_name, obj_color='r'):
-        self._object_map[obj_name], = plt.plot([], [], color=obj_color, marker='o')
+        self._object_map[obj_name], = plt.plot([], [], color=obj_color, marker='o', label=obj_name)
         self._circle_map[obj_name] = plt.Circle((0, 0), 1, color=obj_color, fill=False)
 
         self._objects.append(self._object_map[obj_name])
@@ -61,6 +61,7 @@ class Displayer:
     def animate(self, n_steps, anim_callback):
         def anim_callback_wrapper(t):
             anim_callback(self, t)
+            self._ax.legend()
             return self._objects
 
         anim = FuncAnimation(self._fig,