New Zealand captain Kane Williamson to return from injury for England Test series