changes in weight normalization + else