Minor adjustment to selection feature