small changes to drivetrain code