Add class for data structure