[TASK] Group states by country