Add support for context.

See https://sprint.ly/product/15350/item/289