Ahhh - go with the three speed - unless you're an electronic whizz, in which case ignore me altogether.
The three speeds are purely hydraulic/mechanical, the four speeds are controlled from the ECU - now there are vehicles that use the same AW4 with stand alone TCUs (Transmission Control Units) that you might be able to adapt, but I believe even those require, at the very least, some sort of throttle position input from the ECU, so I'm guessing it's not going to be plug & play.