After a dreadful final session on day two, Sri Lanka had it all to do if they were to avoid the follow-on against New Zealand in Colombo on Tuesday.