Add before/after methods to wrap