code: def set_keyboard_focus(self, move_up, move_down, select):
          
          self.input_focus = StateTypes.KEYBOARD
          self.move_up_button = move_up
          self.move_down_button = move_down
          self.select_button = select
docs: Set the keyboard as the object that controls the menu.
          move_up is from the pygame.KEYS enum that defines what button causes the menu selection to move up.
          move_down is from the pygame.KEYS enum that defines what button causes the menu selection to move down.
          select is from the pygame.KEYS enum that defines what button causes the button to be selected.