Improves example for Transformer Cascade.

Agreed