PTO shaft, you mean the rear shaft of about a 60-70cm length, or the replaceable endpiece ?
If the shaft, drop the pump plate on the bottom of the diff housing (4340 might already have the engine driven pump so you might not find a pump on it) and guide it, as well as the shift sleeve, into place.
Dont forget to drain the oil, or park the tractor nose down into a steep ditch, before dropping the pump plate or youll be covered in oil.