Allow for country_population to be added to all locations as long as county_code is specified.

Author: ExpDev07

app/location/__init__.py

@@ -27,12 +27,13 @@ def country_population(self):
         """
         Gets the population of this location.
         """
+        country_code = self.country_code
 
         # Population data is unavailable for provinces
-        if self.province:
+        if not self.country_code:
             return None
         else:
-            return countrypopulation.get_population_dict()[self.country_code]
+            return countrypopulation.get_population_dict()[country_code]
 
     @property
     def country_code(self):
@@ -106,4 +107,4 @@ def serialize(self, timelines = False):
             }})
 
         # Return the serialized location.
-        return serialized
+        return serialized