Add color to test output