England head coach Trevor Bayliss says they will work with Ben Stokes to get him ready to play in the one day series in New Zealand.