Allow setting multiple context classes.

Also include a minor demo fixup.