Add color function for failed tests