Fixed bug in Lasso