Add colorize functionality