Stokes bags four as NZ struggle to build on strong start