Renamed get_category_count

to get_class_count