diff --git a/TestMode.h b/TestMode.h
index 05f8108..aeb1bb2 100644
--- a/TestMode.h
+++ b/TestMode.h
@@ -1,8 +1,7 @@
 #pragma once
+
 #include <QLabel>
-#include <QPushButton>
 #include <QWidget>
-#include <QVBoxLayout>
 #include "TestModeBtn.h"
 
 class TestMode : public QWidget {
@@ -22,11 +21,11 @@ public:
         this->textcm->setAlignment(Qt::AlignCenter);
         this->mainLayout->addWidget(textcm);
 
-        this->X = new TestModeBtn("X", this);
+        this->X = new TestModeBtn("X", 0, 150, this);
         this->mainLayout->addWidget(X);
-        this->Y = new TestModeBtn("Y", this);
+        this->Y = new TestModeBtn("Y", 0, 300, this);
         this->mainLayout->addWidget(Y);
-        this->Theta = new TestModeBtn("θ", this);
+        this->Theta = new TestModeBtn("θ", -180, 180, this);
         this->mainLayout->addWidget(Theta);
 
         this->go = new QPushButton("Go", this);