Update UnitManager.cs