NN: move of regularization part