Игра на Pygame

import pygame
pygame.init()
 
width = 600
height = 500
w1 = 200
h1 = 10
screen = pygame.display.set_mode((width, height))
screen.fill([197, 0, 128])
pygame.display.flip()
pygame.display.set_caption("Отскок")
clock = pygame.time.Clock()
my_font = pygame.font.SysFont("Arial", 30)
my_font1 = pygame.font.SysFont("Arial", 20)
button_sound = pygame.mixer.Sound("music/button.wav")
ball = pygame.image.load('img\\ball_basket1.png')
bg = pygame.image.load('img\\bg.jpg')
class Button:
    def __init__(self, width, height):
        self.width = width
        self.height = height
        self.inactive_clr = (23, 204, 58)
        self.active_clr = (13, 162, 58)
    def draw(self, x, y ):
        mouse = pygame.mouse.get_pos()
        click = pygame.mouse.get_pressed()
        if x < mouse[0] < x + self.width:
            if y < mouse[1] < y + self.height:
                pygame.draw.rect(screen, self.active_clr, (x, y, self.width, self.height))
 
                if click[0] == 1:
                    w1 = 50
                    h1 = 10
                    pygame.mixer.Sound.play(button_sound)
                    pygame.time.delay(300)
        else:
            pygame.draw.rect(screen, self.inactive_clr, (x, y, self.width, self.height))
FPS = 60
X2 = 500
Y2 = 400
count = 0
r = 5
X1 = 200
Y1 = 250
speed = 10
move_up1 = False
move_down1 = False
move_right1 = False
move_left1 = False
def distance(x1, y1, x2, y2):
    dist = pow((pow(x2-x1,2)+pow(y2-y1,2)),0.5)
    return dist
 
flag_x = 1
flag_y = 1
button = Button(100, 50)
sound = pygame.mixer.Sound('music/Jump.mp3')
game_run = True
while game_run:
    clock.tick(FPS)
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            game_run = False
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_UP:
                move_up1 = True
            if event.key == pygame.K_DOWN:
                move_down1 = True
            if event.key == pygame.K_LEFT:
                move_left1 = True
            if event.key == pygame.K_RIGHT:
                move_right1 = True
 
        if event.type == pygame.KEYUP:
            if event.key == pygame.K_UP:
                move_up1 = False
            if event.key == pygame.K_DOWN:
                move_down1 = False
            if event.key == pygame.K_LEFT:
                move_left1 = False
            if event.key == pygame.K_RIGHT:
                move_right1 = False
    if X1 - r < 0 or X1 + r > width or distance(X1, Y1, X2, Y2) < 30:
        flag_x *= -1
        count += 1
    if Y1 - r < 0 or Y1 + r > height:
 
        flag_y *= -1
        count += 1
 
    if move_up1:
        if Y2 - r > 0:
            Y2 -= speed
    if move_down1:
        if Y2 + r < height:
            Y2 += speed
    if move_right1:
        if X2 + r < width:
            X2 += speed
    if move_left1:
        if X2 - r > 0:
            X2 -= speed
    X1 += speed*flag_x
    Y1 += speed*flag_y
    screen.blit(bg, (0,0))
    screen.blit(ball, (X1, Y1))
    button.draw(20, 100)
    pygame.draw.rect(screen, [255,255,255], [X2, Y2, w1, h1])
    text1 = my_font.render(f'Очки: {count}', True, (255, 255, 255))
    pygame.draw.rect(screen, [0,255, 255], [10, 60, 70, 25])
    text2 = my_font1.render('Легко', True, (255, 0, 0))
    text3 = my_font1.render('Средне', True, (255, 0, 0))
    pygame.draw.rect(screen, [0, 255, 255], [10, 100, 70, 25])
    pygame.draw.rect(screen, [0, 255, 255], [10, 140, 70, 25])
    text4 =  my_font1.render('Сложно', True, (255, 0, 0))
    screen.blit(text4, (15, 140))
    screen.blit(text3, (15,100))
    screen.blit(text2, (15, 60))
    screen.blit(text1, (5, 5))
    pygame.display.flip()
pygame.quit()