TicTacToe in pygame with AI