You can buy gears from the dealer or used diffs from a wrecking yard 01-02 with 4 speed auto. Direct swap. If you buy used, then just bolt in. It's a good time to replace bushings, bearings, seals, etc... Also, ARB locker is nice addition (especially in rear).
More power. Correct speedo...