modifying script lasso_trainer.R