Adding ContextBuilder for wrapping methods