Ben Stokes focused on winning as England return to scene of Wellington thriller