diff --git a/src/behavior_generation_lecture_python/utils/grid_plotting.py b/src/behavior_generation_lecture_python/utils/grid_plotting.py
index 023c378..2823275 100644
--- a/src/behavior_generation_lecture_python/utils/grid_plotting.py
+++ b/src/behavior_generation_lecture_python/utils/grid_plotting.py
@@ -70,7 +70,16 @@ def plot_grid_step(iteration):
                 scaling = np.sqrt(dx**2.0 + dy**2.0) * 2.5
                 dx /= scaling
                 dy /= scaling
-                plt.arrow(x, y, dx, dy)
+                plt.arrow(
+                    x,
+                    y,
+                    dx,
+                    dy,
+                    shape="full",
+                    lw=1.0,
+                    length_includes_head=True,
+                    head_width=0.15,
+                )
         plt.axis("equal")
         plt.xlim([0, columns])
         plt.ylim([0, rows])