Added connection context manager