pygame.error: cannot convert without pygame.display initialized

import pygame
import os
import random
 
 
def load_image(name, colorkey=-1):
    fullname = os.path.join('data', name)
    image = pygame.image.load(fullname).convert()
    if colorkey is not None:
        if colorkey == -1:
            colorkey = image.get_at((0, 0))
        image.set_colorkey(colorkey)
    else:
        image = image.convert_alpha()
    return image
 
 
class Bomb(pygame.sprite.Sprite):
    image = load_image("arrow.png")
 
    def __init__(self, group):
        # НЕОБХОДИМО вызвать конструктор родительского класса Sprite. Это очень важно!!!
        super().__init__(group)
        self.image = Bomb.image
        self.rect = self.image.get_rect()
        self.rect.x = random.randrange(width)
        self.rect.y = random.randrange(height)
 
    def update(self):
        self.rect = self.rect.move(random.randrange(3) - 1, random.randrange(3) - 1)
 
pygame.init()
size = width, height = 501, 501
screen = pygame.display.set_mode(size)
screen.fill((255, 255, 255))
clock = pygame.time.Clock()
fps = 60
running = True
# создадим группу, содержащую все спрайты
all_sprites = pygame.sprite.Group()
 
for _ in range(50):
    Bomb(all_sprites)
 
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
    pygame.display.flip()
    all_sprites.draw(screen)
    all_sprites.update()
 
    clock.tick(fps)
 
pygame.quit()

import pygame
import os
import random
 
pygame.init()
size = width, height = 501, 501
screen = pygame.display.set_mode(size)
 
def load_image(name, colorkey=-1):
    fullname = os.path.join('data', name)
    image = pygame.image.load(fullname).convert()
    if colorkey is not None:
        if colorkey == -1:
            colorkey = image.get_at((0, 0))
        image.set_colorkey(colorkey)
    else:
        image = image.convert_alpha()
    return image
 
 
class Bomb(pygame.sprite.Sprite):
    image = load_image("arrow.png")
 
    def __init__(self, group):
        # НЕОБХОДИМО вызвать конструктор родительского класса Sprite. Это очень важно!!!
        super().__init__(group)
        self.image = Bomb.image
        self.rect = self.image.get_rect()
        self.rect.x = random.randrange(width)
        self.rect.y = random.randrange(height)
 
    def update(self):
        self.rect = self.rect.move(random.randrange(3) - 1, random.randrange(3) - 1)