added DriveTrain to main