Kane Williamson returns as New Zealand announce squad for Test series against England