I've owned a well set up DRZ and currently own a very well set up KLX 331. Of those two I would go with the KLX. They both felt about the same to me on the highway, but the KLX feels much better offroad. With the 331 kit the power feels the same to me as the DRZ, but (subjectively) the KLX feels lighter than the DRZ did.
In your place, I'd go with the KLX.