From d0b2d8ee5e5df4550ee44f0ff1e7b70ce2198699 Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Mon, 18 Mar 2019 14:35:33 -0700
Subject: [PATCH] Small changes to LinearTransformationScene

---
 manimlib/scene/vector_space_scene.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/manimlib/scene/vector_space_scene.py b/manimlib/scene/vector_space_scene.py
index 8b55e733..cd1c8dd4 100644
--- a/manimlib/scene/vector_space_scene.py
+++ b/manimlib/scene/vector_space_scene.py
@@ -287,6 +287,9 @@ class LinearTransformationScene(VectorScene):
         },
         "background_plane_kwargs": {
             "color": GREY,
+            "axis_config": {
+                "stroke_color": LIGHT_GREY,
+            },
             "number_line_config": {
                 "color": GREY,
             },
@@ -358,7 +361,7 @@ class LinearTransformationScene(VectorScene):
         self.add_special_mobjects(self.moving_mobjects, mobject)
 
     def get_unit_square(self, color=YELLOW, opacity=0.3, stroke_width=3):
-        square = Rectangle(
+        square = self.square = Rectangle(
             color=color,
             width=self.plane.get_x_unit_size(),
             height=self.plane.get_y_unit_size(),