Add test/method for rectangle query