Lasso and Group Lasso implementation in Spark