php viser rød, gul eller grøn farve div afhængigt af hvor mange dage der er tilbage?

Hvis jeg var dig ville jeg lave en funktion, der kortlægger dage tilbage til farve:

function traffic_light($days){
  if($days <= 7) return 'red';
  else if($days <= 30) return 'yellow';
  else return 'green';
}

Og rediger derefter koden, der viser linjer, for at bruge den til at vælge farven:

echo "<p>Insurance expires in <font color=\"".traffic_light($days)."\">{$row['expire_date']} day(s)</font></p>";