diff --git a/face_swapper/README.md b/face_swapper/README.md
index 13889e6..7fd5f9e 100644
--- a/face_swapper/README.md
+++ b/face_swapper/README.md
@@ -1,7 +1,7 @@
 Face Swapper
 ============
 
-> Face shape and feature aware identity transfer.
+> Face shape and occlusion aware identity transfer.
 
 ![License](https://img.shields.io/badge/license-ResearchRAIL--MS-red)
 
diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py
index ebf7270..183bb41 100644
--- a/face_swapper/src/dataset.py
+++ b/face_swapper/src/dataset.py
@@ -84,12 +84,13 @@ class AugmentTransform:
 	def compose_transforms() -> albumentations.Compose:
 		return albumentations.Compose(
 		[
-			albumentations.RandomBrightnessContrast(p = 0.3),
+			albumentations.HorizontalFlip(),
 			albumentations.OneOf(
 			[
 				albumentations.MotionBlur(p = 0.1),
 				albumentations.MedianBlur(p = 0.1)
 			], p = 0.3),
+			albumentations.RandomBrightnessContrast(p = 0.3),
 			albumentations.ColorJitter(p = 0.1)
 		])