The 2NZ-FE was never factory mated to any 5 speed automatic gears, so finding one that fits is a challenge by itself. This also will increase the electrical work by fitting parts together that were not meant to go together by Toyota.
In this case your best bet would be to purchase a Belta with a CVT for added fuel economy, sell your current one. There is a 5 gear manual transmission that was mated from the factory would be an option, but I wouldn't mess with a car if it didnt need to.