Add UnitObject class