Add class definition