add classes and colours by country