Improves example for Transformer Cascade.