Finalitzat l'exercici 2 al fitxer *.py. Cal encara afegir-ho al notebook.