New Zealand take 138-run lead in 2nd test vs. Sri Lanka