Shindere/adding test for country

@jzoldak 
Imports should be at the top of the file.