Add array_key_sort helper function