Sri Lanka remain in control