Make Zero a context manager as well (support with)