Add color output to tests