added TODO for single select, and break once we have selected a unit.