Adding W normalization