Included penalty function in fit