Improved box search to work with visual features directly