Add UnitCage class