UnitModel function

The UnitModel class