Did the ELR connector at the back of the injection pump come off? Only affects idle.
Might need new delivery valve seals.
Check balance, check compression.
Is the upper radiator hose too stiff to pinch with your fingers when the engine's idling?
Is there pressure in the cooling system hours after you turn off the engine?