Кнопки в pygame

import pygame
 
class Button:
    def __init__(self,path,text,x,y):
 
        self.size = [190,45] #Размер кнопки
        self.image_button = pygame.image.load(path) #Загружаем изображение исходной кнопки
        self.text = text #Текст кнопки
        self.x = x #Позиция х кнопки
        self.y = y #Позиция у кнопки
        self.rect_button = pygame.Rect(self.x,self.y,self.size[0],self.size[1]) #Прямоугольник для создания коллизии с курсором
        self.rect_image_button = self.image_button.get_rect()
        """Создание кнопки при нажатии на нее"""
        self.image_click = pygame.image.load("images/blue_button03.png")
        self.image_put_on_button = pygame.image.load("images/blue_button01.png")
 
 
    def create_button(self,settings,color):
        """Создает кнопку на экране"""
        text_button = settings.font_text.render(self.text,True,color) #Создаем изображение с текстом
        text_rect = text_button.get_rect() #Возвращаем прямоугольник который занимает текст
        text_rect.center = self.rect_image_button.center #Делаем текст посередине кнопки
        self.image_button.blit(text_button,text_rect)
        self.image_click.blit(text_button,text_rect)
        self.image_put_on_button.blit(text_button,text_rect)
 
    def draw_button(self,screen,control): #Рисует созданную кнопку на экране
       """Рисует кнопку в зависимости от флага,который берется из класса Control"""
       if self.rect_button.collidepoint(control.mouse_x,control.mouse_y) and control.flag == "CLICKED":
 
           screen.blit(self.image_click, (self.x, self.y)) #Если кликнули мышкой то рисуем кнопку нажатой
 
       elif control.flag == "NORMAL":
 
           screen.blit(self.image_button,(self.x,self.y)) #Если ничего не происходит,то рисуем обычную кнопку
 
       elif self.rect_button.collidepoint(control.mouse_x,control.mouse_y):
 
           screen.blit(self.image_put_on_button,(self.x,self.y))

import pygame
from pygame.locals import *
 
class Control:
    def __init__(self):
 
        self.play = True #Переменная для остановки основного цикла игры
        self.mouse_x = 0 #Позиция х курсора
        self.mouse_y = 0 #Позиция у курсора
        self.flag = "NORMAL" #Флаг для отрисовки кнопок
 
    def control_game(self):
        """Все управление игрой"""
        for event in pygame.event.get():
 
            if event.type == QUIT:
                self.play = False
 
            elif event.type == KEYDOWN:
 
                if event.key == K_ESCAPE:
                    self.play = False
 
            elif event.type == MOUSEBUTTONDOWN:
 
                self.mouse_x,self.mouse_y = pygame.mouse.get_pos()
                self.flag = "CLICKED"
 
            elif event.type == MOUSEBUTTONUP:
 
                self.flag = "NORMAL"