add option for color output