add context wrapper