added some unit selection