Ajuste no main para aceitar argumentos