Add compute_ranks function