add color y reorganizado

Se añaden colores de los bordes y se reorganiza el código para una mayor
comprensión