Add get_top_k_categories