Add class_generator