import pygame
pygame.init()
SCREEN_WIDTH = 800
SCREEN_HEIGHT = 600

screen