Add country check