I think what you want is a slip yoke somewhere on the drive shaft.
How did the original 3 speed set up look?