Add shrink_count api