You can use the diesel guts to beef up the capacity of your trans. Add an extra clutch to the front drum to a total of 4 is an easy upgrade. Adding extra OD brake and direct clutches in the OD will also help. A good shift kit with valve body mods to increase line pressure and pressure to the TC lock up clutch is also a good idea. Use a GOOD plate style aftermarket cooler when re-installed to keep the heat in check. The factory cooler set-up through the radiator is not the best to say the least.