added weight diff to the model