about to add in regularization