Add monitoring wrapper