Added the LASSO step