Add category to generator