Update unitManager.cs