Added with_connection