Add class logic