Melhorias no código, inclusão de classes