diff --git a/postgres/config_masking.tdk.yaml b/postgres/config_masking.tdk.yaml
index f4573d8..3ee8ca2 100644
--- a/postgres/config_masking.tdk.yaml
+++ b/postgres/config_masking.tdk.yaml
@@ -1,5 +1,16 @@
 default_config:
-  mode: MASKING
+  mode: KEEP
+
+tables:
+  - table_name_with_schema: "public.customer"
+    transformations:
+      - columns: ["first_name", "last_name", "email"]
+        mode: MASKING
+
+  - table_name_with_schema: "public.staff"
+    transformations:
+      - columns: ["first_name", "last_name", "email", "username", "password"]
+        mode: MASKING
 
 schema_creation_mode: DROP_AND_CREATE
 safety_mode: "RELAXED"
diff --git a/postgres/soda/masked_film_rating_distribution.yml b/postgres/soda/masked_film_rating_distribution.yml
index fd67f6b..c620d09 100644
--- a/postgres/soda/masked_film_rating_distribution.yml
+++ b/postgres/soda/masked_film_rating_distribution.yml
@@ -9,8 +9,8 @@ distribution_reference:
   - 0.194
   - 0.178
   bins:
-  - QZ-54
-  - FX-15
-  - Z
-  - QZ
-  - F
+  - PG
+  - G
+  - NC-17
+  - R
+  - PG-13