Updated to current TransformerKit