Tom Latham and Kane Williamson hit impressive half centuries as New Zealand took the battle to Sri Lanka in the second Test in Galle.