import pygame class Window: def __init__(self, title, size): pygame.init() pygame.display.set_caption(title) self.__surface = pygame.display.set_mode(size=size) def reset(self): self.__surface.fill((0, 0, 0))