Fix up improper reference to DTI matrix (Y->Y0) and make the weight matrix of each round to be the same