import pygame as pg


class GameState:
    def __init__(self, screen):
        self.clock = pg.time.Clock()
        self.screen = screen

    def process_input(self):

    def update(self):
    
    def draw():


class Menu(GameState):
    def __init__(self, screen):
        super().__init__()

class Play(GameState):
    def __init__(self, screen):
        super().__init__()

class End(GameState):
    def __init__(self, screen):
        super().__init__()
        

