Almost have the selecting a Unit working... So close!