class for unit selection has been added