Updating init_country and adding tests
Removing obsolete import
Removing country arg from examples