added comments to UnitController