Как нарисовать сетку 5x5 в Python? Фреймворк Pygame

 import pygame  
  
 
pygame.init()  
  
 
screen = pygame.display.set_mode((500, 500))  
pygame.display.set_caption("Сетка")  
  
 
grid_size = 5  
square_size = 100  
  
  
color = (255, 0, 0)  # red  
  
 
line_thickness = 2  
  
 
for row in range(grid_size):  
    for col in range(grid_size):  
        x = col * square_size  
        y = row * square_size  
        pygame.draw.rect(screen, color, (x, y, square_size, square_size), line_thickness)  
  
 
pygame.display.update()  
  
 
running = True  
while running:  
    for event in pygame.event.get():  
        if event.type == pygame.QUIT:  
            running = False  
  
 
pygame.quit()