self.led_color = social_state.color


class EasterEgg(State):

    label = "Easter Egg"

    def display(self):
        display_bg_and_text(image="images/easter_egg/easter_egg.bmp")
        time.sleep(4.0)  # Wait 4 seconds, then return to main state_manager.
        state_manager.previous_state()


main_menu = MainMenu(NameBadge, SocialBattery, QrCode, Credits, PressStart)

state_manager = StateManager()
state_manager.add(
    main_menu,
    PressStart(),
    Credits(),
    NameBadge(),
    QrCode(),
    SocialBattery(),
    EasterEgg(),
)
state_manager.state = PressStart

while True:
    state_manager.check_for_event()
    time.sleep(0.15)  # Prevent duplicate button presses.