Unit Upgrade

Unit Class has been added