Actualizado el código para python 3