Add class Unit