Adicionar dificuldades

Achei interessante a ideia do joguinho de adivinhação então acrescentei 3 dificuldades para o usuário escolher! Se alguém souber uma forma mais simples de fazer o mesmo. Segue o código: codepen.io

Parte com o código de dificuldade:

//html e alguns códigos
    var quantidadeNumeros = 0;
    var dificuldade = 0;
    while (dificuldade <= 5) {
        var selecionaDificuldade = prompt("Digite a dificuldade que prefere. Somente ''facil'', ''medio'' ou ''dificil''");

        if (selecionaDificuldade == "facil") {
            quantidadeNumeros = 5;
            dificuldade = 10;
            break;
        } 
        if (selecionaDificuldade == "medio") {
            quantidadeNumeros = 3;
            dificuldade = 10;
            break;
        }
        if (selecionaDificuldade == "dificil") {
            quantidadeNumeros = 1;
            dificuldade = 10;
            break;
        }
        if (dificuldade <= 5) {
            alert("Digite somente ''facil'', ''medio'' ou ''dificil'' sem as aspas e em letra minuscula.");
        }
    }    
//código de sorteio
    var segredos = sorteiaNumeros(quantidadeNumeros);
//mais código

Código completo:

<meta charset="utf-8">
<p>Digite um numero de 1 a 10 abaixo:</p>
<input>
<button>Confira se acertou</button>
<script>
    var quantidadeNumeros = 0;
    var dificuldade = 0;
    while (dificuldade <= 5) {

        var selecionaDificuldade = prompt("Digite a dificuldade que prefere. Somente ''facil'', ''medio'' ou ''dificil''");

        if (selecionaDificuldade == "facil") {
            quantidadeNumeros = 5;
            dificuldade = 10;
            break;
        } 
        if (selecionaDificuldade == "medio") {
            quantidadeNumeros = 3;
            dificuldade = 10;
            break;
        }
        if (selecionaDificuldade == "dificil") {
            quantidadeNumeros = 1;
            dificuldade = 10;
            break;
        }
        if (dificuldade <= 5) {
            alert("Digite somente ''facil'', ''medio'' ou ''dificil'' sem as aspas e em letra minuscula.");
        }
    }    

    function sorteia(){
        return Math.round(Math.random()* 10);
    }

    function sorteiaNumeros(quantidade){
        var segredos = [];
        var numero = 1;
        while(numero <= quantidade){
            var numeroAleatorio = sorteia();
            var achou = false;
            if (numeroAleatorio !== 0) {
                for (var posicao = 0; posicao < segredos.length; posicao++) {
                     if (segredos[posicao] == numeroAleatorio) {
                         achou = true;
                         break;
                     }
                }
                if (achou == false) {
                    segredos.push(numeroAleatorio);
                    numero++;
                }
            }
        }
        return segredos;
    }



    var segredos = sorteiaNumeros(quantidadeNumeros);

    //console.log(segredos);
    var input = document.querySelector("input");
    input.focus();
    function verifica() {
        var certo = false;
        for (var posicao = 0; posicao < segredos.length; posicao++) {
            if (input.value == segredos[posicao]) {
                alert("Você acertou!");
                certo = true;
                break;
            }
        }
        if (certo == false) {
            alert("Você errou!");
        }
        input.value = "";
        input.focus();
    }
    var button = document.querySelector("button");
    button.onclick = verifica;
</script>