Improves example for Transformer Cascade.

Already raised as #148. I think that will require a bit of implicit magic though.