Sri Lanka do well in patches with ball and bat but fall 96 runs short in the end