Add colorize option