Herath gives Sri Lanka the advantage