Acomode un poco el código