Pequenas alterações no código.