function disable(square){
    var element = document.getElementById(square);

    if (element.innerHTML == "X" || element.innerHTML  == "O"){
        document.getElementById(square).onclick = false;
    } else if (document.getElementById("restart").click == true) {
        document.getElementById(square).onclick = true;
    }
}