Regularization

fixed 'predict' function