def __init__(self, display): self.display = display self.currentState = 0 self.maxStates = 5 self.states = [ GameState.MenuState(display, self), TrainingState.TrainingState(display, self) ]