added logic to choose country