Worked on fitting gradient, other parts of penalty
add regularization gradient to penalty calculation