There's not much else that'll bolt in that's an auto and an upgrade. If it were me, I'd stay stock or install an AX5 5spd, or search for the hard to find AW4 from a 2.5L XJ or MJ. Rumor has it they existed, it would give you a very strong transmission with a 4spd OD auto.