Virat Kohli marked his return to form in Test cricket with his first half-century of the year, delivering it just when India needed him most in the second innings of the first Test against New Zealand on Friday.