Pequena alteração na classe main.