Refactorizando un pelín el código
Mejorando un poco la interfaz, o eso creo xD