Latham, Williamson pushed Kiwis into lead