Update logic for selection