Southee puts Sri Lanka in trouble