Update DriveTrain.c

Better code!