diff --git a/ElectricLocomotive/ElectricLocomotive/DrawingElectricLocomotive.cs b/ElectricLocomotive/ElectricLocomotive/DrawingElectricLocomotive.cs
index 64921f7..16b18ec 100644
--- a/ElectricLocomotive/ElectricLocomotive/DrawingElectricLocomotive.cs
+++ b/ElectricLocomotive/ElectricLocomotive/DrawingElectricLocomotive.cs
@@ -30,6 +30,7 @@ namespace ProjectElectricLocomotive.DrawingObjects
             Brush additionalBrush = new SolidBrush(electricLocomotive.AdditionalColor);
             Brush bodyColor = new SolidBrush(electricLocomotive.BodyColor);
             g.DrawRectangle(pen, _startPosX + 40, _startPosY + 24, 25, 11);
+            if (electricLocomotive.Compartment)
             g.FillPolygon(additionalBrush, new Point[]
              {
                         new Point(_startPosX + 61, _startPosY + 25),
@@ -39,11 +40,13 @@ namespace ProjectElectricLocomotive.DrawingObjects
                         new Point(_startPosX + 61, _startPosY + 25),
              }
              );
-
-            g.FillRectangle(blackBrush, _startPosX + 30, _startPosY + 15, 20, 5);
-            g.DrawLine(pen, _startPosX + 30, _startPosY + 15, _startPosX + 50, _startPosY + 2);
-            g.DrawLine(pen, _startPosX + 40, _startPosY + 15, _startPosX + 60, _startPosY + 2);
-            base.DrawTransport(g);
+            if (electricLocomotive.Pantograph)
+            {
+                g.FillRectangle(blackBrush, _startPosX + 30, _startPosY + 15, 20, 5);
+                g.DrawLine(pen, _startPosX + 30, _startPosY + 15, _startPosX + 50, _startPosY + 2);
+                g.DrawLine(pen, _startPosX + 40, _startPosY + 15, _startPosX + 60, _startPosY + 2);
+            }
+                base.DrawTransport(g);
         }
         public void SetAdditionalColor(Color color)
         {