an 8.8 from a 96 to 01 exploder would be best. (disk brakes too) also any d30 from the newer xjs would work. its real easy to match a 373 setup like that. you can even go to 410 but those are harder to find. also as for brakes youd benefit from a dual diaphram brake booster and durrango master cylinder. then if you do a lift on it another cheap upgrade is the wj steering/brakes setup.
as for upgrades for the d30 itself. if money isnt an option then you can go with one of the nice eaten or arb lockers. if your more on a budget use an aussie or spartan locker. about the only other things i can think of are chromo shafts... which those can be expensive. i just run the oem shafts with 5-760 spicer ujoints in it.