Uncomment location_count method