Refactored box selection to be based on ROI selection