let take xs ~n =
  take_whilei xs ~f:(fun j _ -> j < n)