Added category to Transformer and selection mechanism