add data augmentation step to the norm method