updated unit selection ui