Add category to story schema