Adding a context argument.