Add color output test