Adding the change of weights as per Professors doc