Add color output option