Add colorization to outputs
Add colorization to output