diff --git a/ProjectElectricLocomotive/ProjectElectricLocomotive/DrawningWheels.cs b/ProjectElectricLocomotive/ProjectElectricLocomotive/DrawningWheels.cs
index d8ef7f3..2e022f5 100644
--- a/ProjectElectricLocomotive/ProjectElectricLocomotive/DrawningWheels.cs
+++ b/ProjectElectricLocomotive/ProjectElectricLocomotive/DrawningWheels.cs
@@ -12,7 +12,7 @@ namespace ProjectElectricLocomotive
         private Wheels amount;
         public void SetAmount(int a)
         {
-            if (a == 2)
+            if (a <= 2)
             {
                 amount = Wheels.Two;
             }
@@ -20,7 +20,7 @@ namespace ProjectElectricLocomotive
             {
                 amount = Wheels.Three;
             }
-            else if (a == 4)
+            else if (a >= 4)
             {
                 amount = Wheels.Four;
             }
@@ -30,7 +30,7 @@ namespace ProjectElectricLocomotive
             Brush wheelsColor = new SolidBrush(Color.Black);
             if (amount == Wheels.Two)
             g.FillEllipse(wheelsColor, _startPosX + 90, _startPosY + 45, 10, 10);
-            g.FillEllipse(wheelsColor, _startPosX + 100, _startPosY + 45, 10, 10);
+            g.FillEllipse(wheelsColor, _startPosX + 140, _startPosY + 45, 10, 10);
 
             if (amount == Wheels.Three)
             {