Update rank_unrank.cc