Made it work with picked_categories