afegi test.py + canvis en l'estructura de control