Add Contextmanager