Added parameters to tweak lambda and to esily split the dataset into training, cv, and test set