add function to colorize output