Limpio un poco el código python