Add optional `context` argument