Implemented unit upgrades