Added keyboard support for game states