Improved coupled-mode theory for nonlinear directional couplers