Collection view for categories, improve model.