diff --git a/__pycache__/move.cpython-310.pyc b/__pycache__/move.cpython-310.pyc
index 6f3a352..ecfd147 100644
Binary files a/__pycache__/move.cpython-310.pyc and b/__pycache__/move.cpython-310.pyc differ
diff --git a/main.py b/main.py
index 292fed3..e0b502d 100644
--- a/main.py
+++ b/main.py
@@ -6,9 +6,9 @@ game.printBoard()
 game2.printBoard()
 ruch = None
 while True:
-    time.sleep(1)
+    #time.sleep(1)
     ruch = game.nextMove(ruch)
     game.printBoard()
-    time.sleep(1)
+    #time.sleep(1)
     ruch = game2.nextMove(ruch)
     game2.printBoard()
\ No newline at end of file
diff --git a/move.py b/move.py
index 3e1f3b6..cfbb03a 100644
--- a/move.py
+++ b/move.py
@@ -27,7 +27,7 @@ class Move:
         self.initial_row, self.initial_column, self.final_row, self.final_column = Move.random_valid_move(self, color)
         self.x=self.board.boardlist[self.initial_row][self.initial_column].piece
         self.board.boardlist[self.initial_row][self.initial_column].piece = None
-        Move.mat(self, self.x, self.final_row, self.final_column)
+        Move.mat(self, self.final_row, self.final_column)
         self.board.boardlist[self.final_row][self.final_column].piece = self.x
         Move.pawn_promotion(self, self.x, self.final_row)
         # update piece
@@ -99,9 +99,10 @@ class Move:
         if x.name == "pawn":
             if (final_row == 0) or (final_row == 7):
                 x.name = "queen"
-    def mat(self, x, final_row, final_column):
+    def mat(self, final_row, final_column):
         if self.board.boardlist[final_row][final_column].has_piece():
             if self.board.boardlist[final_row][final_column].piece.name == 'king':
+                print("mat")
                 exit(0)
         pass
     def check_szach(self, color, initial_row, initial_column, final_row, final_column):
diff --git a/square.py b/square.py
index 863a16e..32832bc 100644
--- a/square.py
+++ b/square.py
@@ -20,8 +20,9 @@ class Square:
         else:
             return False
     @staticmethod
-    def on_board(*args):
-        for arg in args:
-            if arg < 0 or arg > 7:
-                return False
+    def on_board(row, column):
+        if row < 0 or row > 7:
+            return False
+        if column < 0 or column > 7:
+            return False
         return True
\ No newline at end of file