Add __enter__ and __exit__ methods to enable with-statement