Better prompt for unit selection