código refatorado

 - Remover o if
 - Usando apenas uma variável