Enhance validation for category