Implementação de um menu e melhorias no código