Adding logic to change the colour as the counts change