Add correct getLocation function