you can use your 4x4 flywheel (although the 2wd version is lighter and some would thus argue better), water pump, and thermostat housing for your 2wd conversion

in my 2wd engine, I used a 4x4 oil pump and pick up pipe with a 2wd sump and never had any issues