diff --git a/.vscode/settings.json b/.vscode/settings.json
new file mode 100644
index 0000000..db45ec0
--- /dev/null
+++ b/.vscode/settings.json
@@ -0,0 +1,11 @@
+{
+    "workbench.colorCustomizations": {
+        "[One Dark Pro Mix]": {
+            "activityBar.activeBackground": "#393E46",
+            "activityBar.activeBorder": "#E7F6F2",
+            "activityBar.background": "#222831"
+        },
+        "minimap.background": "#00000000",
+        "scrollbar.shadow": "#00000000"
+    }
+}
\ No newline at end of file
diff --git a/engine/train/collect_imgs.py b/engine/train/collect_imgs.py
index 54e47ef..3c8ab0a 100644
--- a/engine/train/collect_imgs.py
+++ b/engine/train/collect_imgs.py
@@ -2,8 +2,15 @@
 
 import cv2
 
-
+# CONSTANT
+ORG = (100, 50)
+FONT = cv2.FONT_HERSHEY_SIMPLEX
+FONTSCALE = 1.3
+COLOR = (0, 255, 0)
+THICKNESS = 3
+LINETYPE = cv2.LINE_AA
 DATA_DIR = './data'
+
 if not os.path.exists(DATA_DIR):
     os.makedirs(DATA_DIR)
 
@@ -21,8 +28,7 @@
     done = False
     while True:
         ret, frame = cap.read()
-        cv2.putText(frame, 'Ready? Press "Q" ! :)', (100, 50), cv2.FONT_HERSHEY_SIMPLEX, 1.3, (0, 255, 0), 3,
-                    cv2.LINE_AA)
+        cv2.putText(frame, 'Ready? Press "Q" ! :)', ORG, FONT, FONTSCALE, COLOR, THICKNESS, LINETYPE)
         cv2.imshow('frame', frame)
         if cv2.waitKey(25) == ord('q'):
             break
diff --git a/engine/train/create_dataset.py b/engine/train/create_dataset.py
index 5e113d0..1993ccc 100644
--- a/engine/train/create_dataset.py
+++ b/engine/train/create_dataset.py
@@ -29,17 +29,12 @@
         if results.multi_hand_landmarks:
             for hand_landmarks in results.multi_hand_landmarks:
                 for i in range(len(hand_landmarks.landmark)):
-                    x = hand_landmarks.landmark[i].x
-                    y = hand_landmarks.landmark[i].y
-
-                    x_.append(x)
-                    y_.append(y)
+                    x_.append(hand_landmarks.landmark[i].x)
+                    y_.append(hand_landmarks.landmark[i].y)
 
                 for i in range(len(hand_landmarks.landmark)):
-                    x = hand_landmarks.landmark[i].x
-                    y = hand_landmarks.landmark[i].y
-                    data_aux.append(x - min(x_))
-                    data_aux.append(y - min(y_))
+                    data_aux.append(hand_landmarks.landmark[i].x - min(x_))
+                    data_aux.append(hand_landmarks.landmark[i].y - min(y_))
 
             data.append(data_aux)
             labels.append(dir_)