[TASK] Limit states listing to country