Improves example for Transformer Cascade.

Fixed.