added the unit class, to be used by the container