Added Lasso support