Added colors, changed output a bit