Minor adjustments to number of wheels checking

Author: peternewell

Source/Orts.Simulation/Simulation/RollingStocks/MSTSWagon.cs

@@ -436,8 +436,8 @@ public virtual void LoadFromWagFile(string wagFilePath)
                 }
             }
 
-            // Ensure Drive Axles is set to a default if no OR value added to WAG file
-            if (WagonNumAxles == 0)
+            // Ensure Drive Axles is set to a default if no OR value added to WAG file, assumes that WagonNumAxles only needs to be set on non locomotive cars
+            if (WagonNumAxles == 0 && WagonType != WagonTypes.Engine)
             {
                 if (MSTSWagonNumWheels != 0 && MSTSWagonNumWheels < 6)
                 {
@@ -2296,11 +2296,16 @@ private void UpdateTrainBaseResistance_DavisLowSpeed()
             // Determine Axle loading of Car
             if (WagonType == WagonTypes.Engine && IsPlayerTrain && Simulator.PlayerLocomotive is MSTSLocomotive locoParameters)
             {
+                // This only takes into account the driven axles for 100% accuracy the non driven axles should also be considered
                 AxleLoadKg = locoParameters.DrvWheelWeightKg / locoParameters.LocoNumDrvAxles;
             }
             else
             {
-                AxleLoadKg = MassKG / WagonNumAxles;
+                // Typically this loop should only be processed when it is a car of some descritption, and therefore it will use the wagon axles as it reference.
+                if (WagonNumAxles > 0)
+                {
+                    AxleLoadKg = MassKG / WagonNumAxles;
+                }
             }
 
             // Calculate the track gradient based on wagon axle loading