mejorando un poco el codigo