Improve heuristic for traffic light color

Now it should work more often!