Implementação em Python para o desafio.