load_data has countries now