Add Weapon and armor model class