Added missing DriveTrain Class