Add regularixation