Update docs with context handling