Added new feature, number of categories