drive shaft assembly