The England captain limped off after clutching his hamstring as New Zealand set the tourists 658 to win and claimed two early wickets.