Add context manager that watches values being set