Add category to Entity model, using enumerize