ENH: context manager + decorator
FIX: doc example