diff --git a/Arduino.ino b/Arduino.ino
index 24d5186..a41a38f 100644
--- a/Arduino.ino
+++ b/Arduino.ino
@@ -1,3 +1,4 @@
+// 3.0.1
 int buttonPin; // Variable to store the pin number for the button
 
 void setup() {
diff --git a/GPDL.py b/GPDL.py
index 7d47a38..6647d79 100644
--- a/GPDL.py
+++ b/GPDL.py
@@ -94,13 +94,13 @@
 
 # Add a 3-second delay
 print("\nPreparing to choose test type...")
+print("\033[33mAttention:\033[0m Do not mix up the modes, otherwise you may damage the gamepad.")
 time.sleep(3)
 
 # Prompt user to select test type (Button test or Stick test)
-print(" ")
-print("Choose test type:")
-print("\033[33mAttention:\033[0m Do not mix up the modes, otherwise you may damage the gamepad.")
-test_type = input("1 for Button test, 2 for Stick test: ")
+print("1 - BUTTON test")
+print("2 - STICK test")
+test_type = input("Enter test type: ")
 
 # Set variables based on selected test type
 if test_type == '1':