England head coach Brendon McCullum has confirmed Ben Stokes will return as captain for the third Test against New Zealand at Trent Bridge.