Move __enter__ and __exit__ to tubes.tube so it's available globally