Update src/drive_train/DriveTrain.cpp