diff --git a/setup.py b/setup.py
index caef839..639d70a 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 from sysconfig import get_config_var, get_paths
 import logging
 
-__version__ = '0.0.4dev'
+__version__ = '0.0.5dev'
 
 
 def get_include():  # TODO
diff --git a/src/krbalancing.cpp b/src/krbalancing.cpp
index 04cbf02..6ddc33b 100644
--- a/src/krbalancing.cpp
+++ b/src/krbalancing.cpp
@@ -1,135 +1,195 @@
 #include "krbalancing.hpp"
 
-kr_balancing::kr_balancing(const  SparseMatrixCol & input){
-    A = input;
-    e.resize(A.rows(),1);
-    e.setOnes();
-    /*Replace zeros with 0.00001 on the main diagonal*/
-    SparseMatrixCol I;
-    I.resize(A.rows(),A.cols());
-    I.setIdentity();
-    I = I*0.00001;
-    A = A + I;
-    rescaled = false;
-}
+// kr_balancing::kr_balancing(const  SparseMatrixCol & input){
+//     A = input;
+//     e.resize(A.rows(),1);
+//     e.setOnes();
+//     /*Replace zeros with 0.00001 on the main diagonal*/
+//     SparseMatrixCol I;
+//     I.resize(A.rows(),A.cols());
+//     I.setIdentity();
+//     I = I*0.00001;
+//     A = A + I;
+//     rescaled = false;
+// }
+
+kr_balancing::kr_balancing(const int & input_rows , const int & input_cols,
+                           const int & input_nnz,
+                           const Eigen::Ref<Eigen::VectorXi> input_indptr,
+                           const Eigen::Ref<Eigen::VectorXi> input_indices,
+                           const Eigen::Ref<Eigen::VectorXd> input_values){
+
+
+            A.resize(input_rows,input_cols);
+            A.reserve(input_nnz);
+            typedef Eigen::Triplet<float> T;
+            std::vector<T> triplets;
+            triplets.reserve(input_nnz);
+
+            // size_t x = input_indptr(0);
+            size_t i = 0;
+            size_t j_start = 0;
+            size_t j_end = 0;
+            std::cout << "start to parse values" << std::endl;
+            while (i < input_indptr.size() - 1) {
+                j_start = input_indptr(i);
+                j_end = input_indptr(i+1);
 
+                // #pragma omp parallel
+                //for (j_start = input_indptr(i); j_start < j_end; j_start++) {
+                while (j_start < j_end) {
+                    // #pragma omp critical
+                    //A.insert(i, input_indices(j_start)) = float(input_values(j_start));
+                    triplets.push_back(T(i, input_indices(j_start),
+                                   float(input_values(j_start))));
+                    j_start++;
+
+                }
+                i++;
+            }
+            std::cout << "end parsing values" << std::endl;
+
+            A.setFromTriplets(triplets.begin(), triplets.end()); //bottleneck
+            //A.makeCompressed();
+            triplets.clear();
+
+
+            e.resize(A.rows(),1);
+            e.setOnes();
+            /*Replace zeros with 0.00001 on the main diagonal*/
+            SparseMatrixCol I;
+            I.resize(A.rows(),A.cols());
+            I.reserve(A.rows());
+            I.setIdentity();
+            I = I*0.00001;
+            A = A + I;
+            rescaled = false;
+}
 
 void kr_balancing::computeKR(){
-  x = e.sparseView();
-  assert(x.isVector());
-  v = x.cwiseProduct(A*x);
-  rk = Eigen::VectorXd::Constant(v.rows(),1) - v; //Vecotr of 1 - v
-  assert(rk.isVector());
-  rho_km1 = rk.conjugate().transpose()*rk;
-  assert(rho_km1.coeff(0,0)>=0);
-  rho_km2 = rho_km1;
-  outer_loop();
+   x = e.sparseView();
+   assert(x.isVector());
+   v = x.cwiseProduct(A*x);
+   rk = Eigen::VectorXd::Constant(v.rows(),1) - v; //Vecotr of 1 - v
+   assert(rk.isVector());
+   rho_km1 = rk.transpose()*rk;
+   assert(rho_km1.coeff(0,0)>=0);
+   rho_km2 = rho_km1;
+   outer_loop();
 }
 
 
 void kr_balancing::outer_loop(){
-  size_t outer_loop_count = 0;
-  double stop_tol = tol*0.5;
-  double eta = etamax;
-  double rt = std::pow(tol,2);
-  double rout = rho_km1.coeff(0,0);
-  double rold = rout;
-  if(fl == 1) std::cout<< 'intermediate convergence statistics is off'<<std::endl;
-  size_t nochange = 0;
-  while(rout > rt){//outer itteration
-    outer_loop_count ++;
-    i=i+1; k=0; y=e.sparseView();
-    innertol = std::max(std::pow(eta,2)*rout,rt);
-    inner_loop();
-    //assert(rho_km1.coeff(0,0) > 0);
-    assert(rho_km1.coeff(0,0) != std::numeric_limits<double>::infinity());
-    x=x.cwiseProduct(y);
-    assert(x.isVector());
-    v=x.cwiseProduct((A*x));
-    rk = Eigen::VectorXd::Constant(v.rows(),1) - v;
-    rho_km1 = rk.conjugate().transpose()*rk;
-    //assert(rho_km1.coeff(0,0) > 0);
-    assert(rho_km1.coeff(0,0) != std::numeric_limits<double>::infinity());
-    rout = rho_km1.coeff(0,0);
-    MVP = MVP + k + 1;
-    //Update inner iteration stopping criterion.
-    double rat = rout/rold; rold = rout; double res_norm = std::sqrt(rout);
-    double eta_o = eta; eta = g*rat;
-    if(g*std::pow(eta_o,2) > 0.1){
-        eta = std::max(eta,g*std::pow(eta_o,2));
-    }
-    eta = std::max(std::min(eta,etamax),stop_tol/res_norm);
-    if(fl == 1){
-        res.push_back(res_norm);
-    }
-    if(outer_loop_count %50 ==0) std::cout << "outer loop number "<< outer_loop_count <<std::endl;
-  }//End of outer loop
-  //    if (nochange >= 100) {
-  //       std::cout << "nochange >=100" <<std::endl;
-  //       return 0;
-  //    }else{
-  //    return &x;
-  //    }
+   size_t outer_loop_count = 0;
+   double stop_tol = tol*0.5;
+   double eta = etamax;
+   double rt = tol*tol;
+   double rout = rho_km1.coeff(0,0);
+   double rold = rout;
+   if(fl == 1) std::cout<< "intermediate convergence statistics is off" <<std::endl;
+   size_t nochange = 0;
+   while(rout > rt){//outer itteration
+     outer_loop_count ++;
+     i=i+1; k=0; y=e.sparseView();
+     innertol = std::max(eta*eta*rout,rt);
+     inner_loop();
+     //assert(rho_km1.coeff(0,0) > 0);
+     assert(rho_km1.coeff(0,0) != std::numeric_limits<double>::infinity());
+     x=x.cwiseProduct(y);
+     assert(x.isVector());
+     v=x.cwiseProduct((A*x));
+     rk = Eigen::VectorXd::Constant(v.rows(),1) - v;
+     rho_km1 = rk.transpose()*rk;
+     //assert(rho_km1.coeff(0,0) > 0);
+     assert(rho_km1.coeff(0,0) != std::numeric_limits<double>::infinity());
+     rout = rho_km1.coeff(0,0);
+     MVP = MVP + k + 1;
+     //Update inner iteration stopping criterion.
+     double rat = rout/rold; rold = rout; double res_norm = std::sqrt(rout);
+     double eta_o = eta; eta = g*rat;
+     if(g*eta_o*eta_o > 0.1){
+          eta = std::max(eta,g*eta_o*eta_o);
+     }
+     eta = std::max(std::min(eta,etamax),stop_tol/res_norm);
+     if(fl == 1){
+         res.push_back(res_norm);
+     }
+     if(outer_loop_count %50 ==0) std::cout << "outer loop number "<< outer_loop_count <<std::endl;
+     if(outer_loop_count %100 ==0){
+       std::cout << x <<std::endl;
+     }
+     if(outer_loop_count %300 ==0){
+       std::cout << x <<std::endl;
+       exit(0);
+     }
+   }//End of outer loop
+   //    if (nochange >= 100) {
+   //       std::cout << "nochange >=100" <<std::endl;
+   //       return 0;
+   //    }else{
+   //    return &x;
+   //    }
 }
 
 
 void kr_balancing::inner_loop(){
-  size_t inner_loop_count = 0;
-  while (rho_km1.coeff(0,0) > innertol){ // Inner itteration (conjugate gradient method)
-    inner_loop_count ++;
-    k++;
-    if(k == 1){
-      Z = rk.cwiseQuotient(v);
-      p=Z;
-      rho_km1 = rk.conjugate().transpose()*Z;
-    }else{
-      Eigen::VectorXd beta=rho_km1.cwiseQuotient(rho_km2);
-      p = Z + (beta(0)*p);
-    }
-    //Update search direction efficiently.
-    SparseMatrixCol w=x.cwiseProduct(A*(x.cwiseProduct(p))) + v.cwiseProduct(p);
-    SparseMatrixCol alpha_mat = rho_km1.cwiseQuotient(p.conjugate().transpose()*w);
-    double alpha = alpha_mat.coeff(0,0);
-    Eigen::VectorXd ap = alpha * p;
-    //Test distance to boundary of cone.
-    Eigen::VectorXd ynew = y + ap;
-    if(ynew.minCoeff() <= delta){
-      if(delta == 0) break;
-      Eigen::Matrix<bool, Eigen::Dynamic , Eigen::Dynamic> ind_helper = (ap.array()<0);
-      Eigen::VectorXi ind = Eigen::VectorXi::LinSpaced(ind_helper.size(),0,ind_helper.size()-1);
-      ind.conservativeResize(std::stable_partition(
-      ind.data(), ind.data()+ind.size(), [&ind_helper](int i){return ind_helper(i);})-ind.data());
-      Eigen::VectorXd y_copy;
-      y_copy.resize(ind.size());
-      for(size_t i = 0; i < ind.size(); i++){
-        y_copy(i) = (delta - y(ind(i)))/ap(ind(i));
-      }
-      double gamma = y_copy.minCoeff();
-      y = y + gamma*ap;
-      break;
-    }
-    if(ynew.maxCoeff() >= Delta){
-      Eigen::Matrix<bool, Eigen::Dynamic , Eigen::Dynamic> ind_helper = (ynew.array() > Delta);
-      Eigen::VectorXi ind = Eigen::VectorXi::LinSpaced(ind_helper.size(),0,ind_helper.size()-1);
-      ind.conservativeResize(std::stable_partition(
-      ind.data(), ind.data()+ind.size(), [&ind_helper](int i){return ind_helper(i);})-ind.data());
-      Eigen::VectorXd y_copy;
-      y_copy.resize(ind.size());
-      for(size_t i = 0; i < ind.size(); i++){
-        y_copy(i) = (Delta - y(ind(i)))/ap(ind(i));
-      }
-      double gamma = y_copy.minCoeff();
-      y = y + (gamma*ap);
-      break;
-    }
-    y = ynew;
-    rk = rk - (alpha*w); rho_km2 = rho_km1;
-    Z = rk.cwiseQuotient(v); rho_km1 = rk.conjugate().transpose()*Z;
+   size_t inner_loop_count = 0;
+   while (rho_km1.coeff(0,0) > innertol){ // Inner itteration (conjugate gradient method)
+     inner_loop_count ++;
+     k++;
+     if(k == 1){
+       Z = rk.cwiseQuotient(v);
+       p=Z;
+       rho_km1 = rk.conjugate().transpose()*Z;
+     }else{
+       Eigen::VectorXd beta=rho_km1.cwiseQuotient(rho_km2);
+       p = Z + (beta(0)*p);
+     }
+     //Update search direction efficiently.
+     SparseMatrixCol w=x.cwiseProduct(A*(x.cwiseProduct(p))) + v.cwiseProduct(p);
+     SparseMatrixCol alpha_mat = rho_km1.cwiseQuotient(p.conjugate().transpose()*w);
+     double alpha = alpha_mat.coeff(0,0);
+     Eigen::VectorXd ap = alpha * p;
+     //Test distance to boundary of cone.
+     Eigen::VectorXd ynew = y + ap;
+     if(ynew.minCoeff() <= delta){
+       if(delta == 0) break;
+       Eigen::Matrix<bool, Eigen::Dynamic , Eigen::Dynamic> ind_helper = (ap.array()<0);
+       Eigen::VectorXi ind = Eigen::VectorXi::LinSpaced(ind_helper.size(),0,ind_helper.size()-1);
+       ind.conservativeResize(std::stable_partition(
+       ind.data(), ind.data()+ind.size(), [&ind_helper](int i){return ind_helper(i);})-ind.data());
+       Eigen::VectorXd y_copy;
+       y_copy.resize(ind.size());
+       for(size_t i = 0; i < ind.size(); i++){
+         y_copy(i) = (delta - y(ind(i)))/ap(ind(i));
+       }
+       double gamma = y_copy.minCoeff();
+       y = y + gamma*ap;
+       break;
+     }
+     if(ynew.maxCoeff() >= Delta){
+       Eigen::Matrix<bool, Eigen::Dynamic , Eigen::Dynamic> ind_helper = (ynew.array() > Delta);
+       Eigen::VectorXi ind = Eigen::VectorXi::LinSpaced(ind_helper.size(),0,ind_helper.size()-1);
+       ind.conservativeResize(std::stable_partition(
+       ind.data(), ind.data()+ind.size(), [&ind_helper](int i){return ind_helper(i);})-ind.data());
+       Eigen::VectorXd y_copy;
+       y_copy.resize(ind.size());
+       for(size_t i = 0; i < ind.size(); i++){
+         y_copy(i) = (Delta - y(ind(i)))/ap(ind(i));
+       }
+       double gamma = y_copy.minCoeff();
+       y = y + (gamma*ap);
+       break;
+     }
+     y = ynew;
+     rk = rk - (alpha*w); rho_km2 = rho_km1;
+     Z = rk.cwiseQuotient(v); rho_km1 = rk.conjugate().transpose()*Z;
   }//End of the inner 'while'
 }
 
 
 void kr_balancing::compute_normalised_matrix(bool & rescale){
+  
   assert(A.rows() == A.cols());
   if(rescale ==true && rescaled == false){
     rescale_norm_vector();
@@ -144,13 +204,15 @@ void kr_balancing::compute_normalised_matrix(bool & rescale){
       it.valueRef() = it.value()*x.coeff(it.row(),0)*x.coeff(it.col(),0);
     }
   }
+  
 }
 
 
 //Rescaling the normalisation factor (x)
 void kr_balancing::rescale_norm_vector(){
-  double original_sum = 0.0;
-  double norm_vector_sum = 0.0;
+
+  float original_sum = 0.0;
+  float norm_vector_sum = 0.0;
   assert(A.rows() == A.cols());
   A = SparseMatrixCol(A.triangularView<Eigen::Upper>());
 
@@ -170,31 +232,38 @@ void kr_balancing::rescale_norm_vector(){
 
   std::cout << "normalisation factor is "<< std::sqrt(norm_vector_sum/original_sum) <<std::endl;
   x /= std::sqrt(norm_vector_sum/original_sum);
-
+  
 }
 
 
 const SparseMatrixCol* kr_balancing::get_normalised_matrix(bool & rescale){
-  compute_normalised_matrix(rescale);
-  return &A;
+   compute_normalised_matrix(rescale);
+   return &A;
 }
 
 
 const SparseMatrixCol* kr_balancing::get_normalisation_vector(bool & rescale){
+  
   if(rescale ==true && rescaled == false){
     rescale_norm_vector();
     rescaled = true;
   }
+  
   return &x;
 }
 
 
 PYBIND11_MODULE(krbalancing, m) {
   py::class_<kr_balancing>(m, "kr_balancing")
-    .def(py::init< const SparseMatrixCol & >())
+    .def(py::init< const int &, const int &, const int &,
+                   const Eigen::Ref<Eigen::VectorXi>,
+                   const Eigen::Ref<Eigen::VectorXi>,
+                   const Eigen::Ref<Eigen::VectorXd> >())
     .def("computeKR", &kr_balancing::computeKR)
-    .def("get_normalisation_vector",&kr_balancing::get_normalisation_vector, py::return_value_policy::reference_internal)
-    .def("get_normalised_matrix",&kr_balancing::get_normalised_matrix, py::return_value_policy::reference_internal);
+    .def("get_normalisation_vector",&kr_balancing::get_normalisation_vector,
+         py::return_value_policy::reference_internal, py::arg().noconvert())
+    .def("get_normalised_matrix",&kr_balancing::get_normalised_matrix,
+         py::return_value_policy::reference_internal, py::arg().noconvert());
 
 }
 
diff --git a/src/krbalancing.hpp b/src/krbalancing.hpp
index 7a5f0e1..65fba53 100644
--- a/src/krbalancing.hpp
+++ b/src/krbalancing.hpp
@@ -22,12 +22,17 @@
 #include <omp.h>
 #include <limits>
 namespace py = pybind11;
-using SparseMatrixCol = Eigen::SparseMatrix<double,Eigen::ColMajor>;
+using SparseMatrixCol = Eigen::SparseMatrix<double, Eigen::ColMajor>;
 size_t num_threads = 10; //TODO add it as an argument to be set by user
 
 class kr_balancing{
      public:
-       kr_balancing(const  SparseMatrixCol & input);
+       kr_balancing(const int & input_rows , const int & input_cols,
+                    const int & input_nnz,
+                    const Eigen::Ref<Eigen::VectorXi> input_outer,
+                    const Eigen::Ref<Eigen::VectorXi> input_inner,
+                    const Eigen::Ref<Eigen::VectorXd> input_values);
+       ~kr_balancing(){}
        void computeKR();
        void outer_loop();
        void inner_loop();