<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def set_keyboard_focus(self, move_up, move_down, select):
 """ 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. """

        self.input_focus = StateTypes.KEYBOARD
        self.move_up_button = move_up
        self.move_down_button = move_down
        self.select_button = select