added the ability to select units