import pygame width, height = 640, 640 rows, cols = 10, 10 size = width//cols yellow = (216,178,0) green= (103,178,0) red = (255,0,0)