added code for drive train we're making