Traduzido o exercício 2 de java para python