coding style
Changed country migration to work with migrate function