Limpando o código e usando o array

Fiz uma limpeza no código e guardando tudo em array