Black Caps take control

New Zealand carved out a 338-run lead over England following an absorbing day three of the second Test at Headingley.