Add class stuff