let color = 'X'; // or 'O'
    button.innerText = color;
    button.setAttribute('data-state', color);