Before you separate the shaft from the trans mark the drive shaft yoke and trans output shaft, you want this to go back EXACTLY the same way you took it apart. Note the arrow on the flex joint points at a flange.