added drive train code