diff --git a/environments/factory/assets/agent.png b/environments/factory/assets/agents/agent.png
similarity index 100%
rename from environments/factory/assets/agent.png
rename to environments/factory/assets/agents/agent.png
diff --git a/environments/factory/assets/agent1.png b/environments/factory/assets/agents/agent1.png
similarity index 100%
rename from environments/factory/assets/agent1.png
rename to environments/factory/assets/agents/agent1.png
diff --git a/environments/factory/assets/agents/agent1violation.png b/environments/factory/assets/agents/agent1violation.png
new file mode 100644
index 0000000..119a0d3
Binary files /dev/null and b/environments/factory/assets/agents/agent1violation.png differ
diff --git a/environments/factory/assets/agent2.png b/environments/factory/assets/agents/agent2.png
similarity index 100%
rename from environments/factory/assets/agent2.png
rename to environments/factory/assets/agents/agent2.png
diff --git a/environments/factory/assets/agent3.png b/environments/factory/assets/agents/agent3.png
similarity index 100%
rename from environments/factory/assets/agent3.png
rename to environments/factory/assets/agents/agent3.png
diff --git a/environments/factory/assets/agent4.png b/environments/factory/assets/agents/agent4.png
similarity index 100%
rename from environments/factory/assets/agent4.png
rename to environments/factory/assets/agents/agent4.png
diff --git a/environments/factory/assets/agent5.png b/environments/factory/assets/agents/agent5.png
similarity index 100%
rename from environments/factory/assets/agent5.png
rename to environments/factory/assets/agents/agent5.png
diff --git a/environments/factory/assets/agent6.png b/environments/factory/assets/agents/agent6.png
similarity index 100%
rename from environments/factory/assets/agent6.png
rename to environments/factory/assets/agents/agent6.png
diff --git a/environments/factory/renderer.py b/environments/factory/renderer.py
index 34c6879..6ee3a2f 100644
--- a/environments/factory/renderer.py
+++ b/environments/factory/renderer.py
@@ -28,7 +28,7 @@ class Renderer:
         self.screen_size = (grid_h*cell_size, grid_w*cell_size)
         self.screen = pygame.display.set_mode(self.screen_size)
         self.clock = pygame.time.Clock()
-        assets = list((Path(__file__).parent / 'assets').glob('*.png'))
+        assets = list((Path(__file__).parent / 'assets').rglob('*.png'))
         self.assets = {path.stem: self.load_asset(str(path), 0.97) for path in assets}
         self.fill_bg()