Update DriveTrain.cpp

Changes to match include file.