improved color output handling